Update app.py

app.py

@@ -48,14 +48,14 @@ embeddings_dict = {}
 with safe_open("unaestheticXLv31.safetensors", framework="pt") as f:
     for k in f.keys():
         embeddings_dict[k] = f.get_tensor(k)
+pipe.text_encoder.resize_token_embeddings(len(pipe.tokenizer),pad_to_multiple_of=128)
+pipe.text_encoder_2.resize_token_embeddings(len(pipe.tokenizer),pad_to_multiple_of=128)
 for i in range(len(embeddings_dict["clip_l"])):
     token = f"sksd{chr(token_num)}"
     token_num+=1
     unaestheticXLv31 += token
     pipe.tokenizer.add_tokens(token)
     token_id = pipe.tokenizer.convert_tokens_to_ids(token)
-    pipe.text_encoder.resize_token_embeddings(len(pipe.tokenizer),pad_to_multiple_of=128)
-    pipe.text_encoder_2.resize_token_embeddings(len(pipe.tokenizer),pad_to_multiple_of=128)
     pipe.text_encoder.get_input_embeddings().weight.data[token_id] = embeddings_dict["clip_l"][i]
     pipe.text_encoder_2.get_input_embeddings().weight.data[token_id] = embeddings_dict["clip_g"][i]
 
@@ -64,14 +64,14 @@ embeddings_dict = {}
 with safe_open("unaestheticXLv1.safetensors", framework="pt") as f:
     for k in f.keys():
         embeddings_dict[k] = f.get_tensor(k)
+pipe.text_encoder.resize_token_embeddings(len(pipe.tokenizer),pad_to_multiple_of=128)
+pipe.text_encoder_2.resize_token_embeddings(len(pipe.tokenizer),pad_to_multiple_of=128)
 for i in range(len(embeddings_dict["clip_l"])):
     token = f"sksd{chr(token_num)}"
     token_num+=1
     unaestheticXLv1 += token
     pipe.tokenizer.add_tokens(token)
     token_id = pipe.tokenizer.convert_tokens_to_ids(token)
-    pipe.text_encoder.resize_token_embeddings(len(pipe.tokenizer),pad_to_multiple_of=128)
-    pipe.text_encoder_2.resize_token_embeddings(len(pipe.tokenizer),pad_to_multiple_of=128)
     pipe.text_encoder.get_input_embeddings().weight.data[token_id] = embeddings_dict["clip_l"][i]
     pipe.text_encoder_2.get_input_embeddings().weight.data[token_id] = embeddings_dict["clip_g"][i]
 
@@ -80,14 +80,15 @@ embeddings_dict = {}
 with safe_open("unaestheticXLv13.safetensors", framework="pt") as f:
     for k in f.keys():
         embeddings_dict[k] = f.get_tensor(k)
+
+pipe.text_encoder.resize_token_embeddings(len(pipe.tokenizer),pad_to_multiple_of=128)
+pipe.text_encoder_2.resize_token_embeddings(len(pipe.tokenizer),pad_to_multiple_of=128)
 for i in range(len(embeddings_dict["clip_l"])):
     token = f"sksd{chr(token_num)}"
     unaestheticXLv13 += token
     token_num+=1
     pipe.tokenizer.add_tokens(token)
     token_id = pipe.tokenizer.convert_tokens_to_ids(token)
-    pipe.text_encoder.resize_token_embeddings(len(pipe.tokenizer),pad_to_multiple_of=128)
-    pipe.text_encoder_2.resize_token_embeddings(len(pipe.tokenizer),pad_to_multiple_of=128)
     pipe.text_encoder.get_input_embeddings().weight.data[token_id] = embeddings_dict["clip_l"][i]
     pipe.text_encoder_2.get_input_embeddings().weight.data[token_id] = embeddings_dict["clip_g"][i]