Update app.py

app.py

@@ -27,8 +27,12 @@ def load_learned_embed_in_clip(learned_embeds_path, text_encoder, tokenizer, tok
 
   # cast to dtype of text_encoder
   dtype = text_encoder.get_input_embeddings().weight.dtype
-  embeds.to(dtype)
-
+  #If embed invalid, skip
+  try:
+    embeds.to(dtype)
+  except:
+    continue
+    
   # add the token in tokenizer
   token = token if token is not None else trained_token
   num_added_tokens = tokenizer.add_tokens(token)