Update app.py

app.py

@@ -6,7 +6,8 @@ from peft import PeftModel, LoraConfig
 import os
 
 def get_lora_sd_pipeline(
-    ckpt_dir='./lora_logos', 
+    #ckpt_dir='./lora_logos', 
+    ckpt_dir='./lora_man_animestyle',
     base_model_name_or_path=None, 
     dtype=torch.float16, 
     adapter_name="default"
@@ -53,10 +54,13 @@ def align_embeddings(prompt_embeds, negative_prompt_embeds):
            torch.nn.functional.pad(negative_prompt_embeds, (0, 0, 0, max_length - negative_prompt_embeds.shape[1]))
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model_id_default = "CompVis/stable-diffusion-v1-4"
+#model_id_default = "CompVis/stable-diffusion-v1-4"
+model_id_default = "stable-diffusion-v1-5/stable-diffusion-v1-5"
 torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
 
-pipe_default = get_lora_sd_pipeline(ckpt_dir='./lora_logos', base_model_name_or_path=model_id_default, dtype=torch_dtype).to(device)
+#pipe_default = get_lora_sd_pipeline(ckpt_dir='./lora_logos', base_model_name_or_path=model_id_default, dtype=torch_dtype).to(device)
+pipe_default = get_lora_sd_pipeline(ckpt_dir='./lora_man_animestyle', base_model_name_or_path=model_id_default, dtype=torch_dtype).to(device)
+
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
@@ -67,7 +71,8 @@ def infer(
     width=512, 
     height=512, 
     num_inference_steps=20, 
-    model_id='CompVis/stable-diffusion-v1-4', 
+    #model_id='CompVis/stable-diffusion-v1-4',
+    model_id='stable-diffusion-v1-5/stable-diffusion-v1-5', 
     seed=42, 
     guidance_scale=7.0, 
     lora_scale=0.5,
@@ -117,7 +122,8 @@ with gr.Blocks(css=css) as demo:
             model_id = gr.Textbox(
                 label="Model ID",
                 max_lines=1,
-                placeholder="Enter model id like 'CompVis/stable-diffusion-v1-4'",
+                #placeholder="Enter model id like 'CompVis/stable-diffusion-v1-4'",
+                placeholder="Enter model id like 'stable-diffusion-v1-5/stable-diffusion-v1-5'",
                 value=model_id_default
             )