Update app.py

app.py

@@ -27,12 +27,7 @@ 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
-  #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)
@@ -83,8 +78,11 @@ for model in models_list:
       file.close()
       images.append(f"{model_id}/{i}.jpeg")
   model_content["images"] = images
-
-  learned_token = load_learned_embed_in_clip(f"{model_id}/learned_embeds.bin", pipe.text_encoder, pipe.tokenizer, token_name)
+  #if token cannot be loaded, skip it
+  try:
+    learned_token = load_learned_embed_in_clip(f"{model_id}/learned_embeds.bin", pipe.text_encoder, pipe.tokenizer, token_name)
+  except: 
+    continue
   model_content["token"] = learned_token
   models.append(model_content)