small fixes

Author: divamgupta

backends/model_converter/convert_model.py

@@ -33,7 +33,7 @@
 extra_keys = ['temb_coefficients_fp32' , 'causal_mask' , 'aux_output_conv.weight' , 'aux_output_conv.bias', 'alphas_cumprod']
 
 for k in torch_weights['state_dict']:
-    if k not in SD_SHAPES:
+    if k not in SD_SHAPES and k not in extra_keys:
         continue
     np_arr = torch_weights['state_dict'][k]
     key_bytes = np_arr.tobytes()

electron_app/src/StableDiffusion.vue

@@ -152,11 +152,13 @@ export default {
             if(!this.is_input_avail)
                 return;
             let tokens = [49406].concat((get_tokens(prompt_params.prompt))).concat([49407])
+            tokens.filter(n => n != null && n != undefined)
             prompt_params.prompt_tokens = tokens;
 
             if(prompt_params.negative_prompt)
             {
                 let tokens2 = [49406].concat((get_tokens(prompt_params.negative_prompt))).concat([49407])
+                tokens2.filter(n => n != null && n != undefined)
                 prompt_params.negative_prompt_tokens = tokens2
             }