Update app.py

app.py

@@ -27,57 +27,9 @@ pipe=pipe.to("cuda")
 pipe.enable_freeu(s1=0.6, s2=0.4, b1=1.1, b2=1.2)
 #pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
 
-token_num=65
-
-unaestheticXLv31=""
-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.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]
-
-unaestheticXLv1=""
-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.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]
-
-unaestheticXLv13=""
-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.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]
-
+state_dict = load_file("unaestheticXLv31.safetensors")
+pipe.load_textual_inversion(state_dict["clip_g"], token="unaestheticXLv31", text_encoder=pipe.text_encoder_2, tokenizer=pipe.tokenizer_2)
+pipe.load_textual_inversion(state_dict["clip_l"], token="unaestheticXLv31", text_encoder=pipe.text_encoder, tokenizer=pipe.tokenizer)
 
 compel = Compel(tokenizer=[pipe.tokenizer, pipe.tokenizer_2] , 
                 text_encoder=[pipe.text_encoder, pipe.text_encoder_2], 
@@ -114,7 +66,7 @@ def auto_prompt_correction(prompt_ui,neg_prompt_ui,disable_auto_prompt_correctio
 
     if(prompt=="" and neg_prompt==""):
         prompt="1girl++, smile--, brown bob+++ hair, brown eyes, sunflowers, sky, transparent++"
-        neg_prompt=f"({unaestheticXLv31})---, photo, deformed, realism, disfigured, low contrast, bad hand"
+        neg_prompt=f"unaestheticXLv31---, photo, deformed, realism, disfigured, low contrast, bad hand"
         return prompt, neg_prompt
 
     splited_prompt=prompt.replace(","," ").replace("_"," ").replace("+"," ").split(" ")
@@ -123,14 +75,14 @@ def auto_prompt_correction(prompt_ui,neg_prompt_ui,disable_auto_prompt_correctio
     for word in human_words:
         if( word in splited_prompt):
             prompt=f"anime artwork, anime style, {prompt}"
-            neg_prompt=f"({unaestheticXLv31})---,{neg_prompt}, photo, deformed, realism, disfigured, low contrast, bad hand" 
+            neg_prompt=f"unaestheticXLv31---,{neg_prompt}, photo, deformed, realism, disfigured, low contrast, bad hand" 
             return prompt, neg_prompt
             
     animal_words=["cat","dog","bird","pigeon","rabbit","bunny","horse"]
     for word in animal_words:
         if( word in splited_prompt):
             prompt=f"anime style, a {prompt}, 4k, detailed"
-            neg_prompt=f"{neg_prompt},({unaestheticXLv31})---" 
+            neg_prompt=f"{neg_prompt},unaestheticXLv31---" 
             return prompt, neg_prompt
             
     background_words=["mount fuji","mt. fuji","building", "buildings", "tokyo", "kyoto", "nara", "shibuya", "shinjuku"]