Update app.py

app.py

@@ -14,16 +14,25 @@ from safetensors.torch import load_file
 
 model_id = 'aipicasso/emi'
 auth_token=os.environ["ACCESS_TOKEN"]
+adapter_id = "latent-consistency/lcm-lora-sdxl"
 
 #scheduler=DPMSolverMultistepScheduler()
 
-pipe = StableDiffusionXLPipeline.from_pretrained(
-  model_id,
-  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
-  use_auth_token=auth_token)
+#pipe = StableDiffusionXLPipeline.from_pretrained(
+#  model_id,
+#  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
+#  use_auth_token=auth_token)
 
+pipe = AutoPipelineForText2Image.from_pretrained(
+    model_id, 
+    torch_dtype=torch.float16,
+    use_auth_token=auth_token
+)
 
 pipe=pipe.to("cuda")
+pipe.load_lora_weights(adapter_id)
+pipe.fuse_lora()
+
 #pipe.enable_freeu(s1=0.9, s2=0.2, b1=1.3, b2=1.4)
 #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)
@@ -118,7 +127,7 @@ with gr.Blocks(css=css) as demo:
         f"""
             <div class="main-div">
               <div>
-                <h1>Emi Demo</h1>
+                <h1>Emi+LCM-LoRA Demo</h1>
                 <!--
                 <h2>
                 Other Demos: 
@@ -165,8 +174,8 @@ with gr.Blocks(css=css) as demo:
               disable_auto_prompt_correction = gr.Checkbox(label="Disable auto prompt corretion.")
                 
               with gr.Row():
-                guidance = gr.Slider(label="Guidance scale", value=7.5, maximum=25)
-                steps = gr.Slider(label="Steps", value=20, minimum=2, maximum=30, step=1)
+                guidance = gr.Slider(label="Guidance scale", value=3, maximum=25)
+                steps = gr.Slider(label="Steps", value=8, minimum=2, maximum=30, step=1)
     
               seed = gr.Slider(0, 2147483647, label='Seed (0 = random)', value=0, step=1)