Update app.py

Добавлена возможность выбора LoRA PussInBoots

app.py

@@ -61,7 +61,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 model_id_default = "sd-legacy/stable-diffusion-v1-5"
 model_dropdown = ['stabilityai/sdxl-turbo', 'CompVis/stable-diffusion-v1-4', 'sd-legacy/stable-diffusion-v1-5'  ]
 
-model_lora_default = "lora_lady_and_cats_logos"
+model_lora_default = "lora_pussinboots_logos"
 model_lora_dropdown = ['lora_lady_and_cats_logos', 'lora_pussinboots_logos'  ]
 
 if torch.cuda.is_available():
@@ -109,8 +109,8 @@ def infer(
         prompt_embeds, negative_prompt_embeds = align_embeddings(prompt_embeds, negative_prompt_embeds)
     else:
         # добавляем lora
-        pipe = get_lora_sd_pipeline(ckpt_dir='./lora_lady_and_cats_logos', base_model_name_or_path=model_id_default, dtype=torch_dtype).to(device)
-        #pipe = get_lora_sd_pipeline(ckpt_dir='./'+model_lora_id, base_model_name_or_path=model_id_default, dtype=torch_dtype).to(device)
+        #pipe = get_lora_sd_pipeline(ckpt_dir='./lora_lady_and_cats_logos', base_model_name_or_path=model_id_default, dtype=torch_dtype).to(device)
+        pipe = get_lora_sd_pipeline(ckpt_dir='./'+model_lora_id, base_model_name_or_path=model_id_default, dtype=torch_dtype).to(device)
         prompt_embeds = process_prompt(prompt, pipe.tokenizer, pipe.text_encoder)
         negative_prompt_embeds = process_prompt(negative_prompt, pipe.tokenizer, pipe.text_encoder)
         prompt_embeds, negative_prompt_embeds = align_embeddings(prompt_embeds, negative_prompt_embeds)