change space for hw5 #2

app.py

@@ -5,7 +5,7 @@ import numpy as np
 import torch
 
 # import spaces #[uncomment to use ZeroGPU]
-from diffusers import DiffusionPipeline
+from diffusers import StableDiffusionPipeline
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 # model_repo_id = "stabilityai/sdxl-turbo"  # Replace to the model you would like to use
@@ -50,7 +50,7 @@ def infer(
 
     generator = torch.Generator().manual_seed(seed)
 
-    pipe = DiffusionPipeline.from_pretrained(
+    pipe = StableDiffusionPipeline.from_pretrained(
         model_id,
         torch_dtype=torch_dtype,
         requires_safety_checker=False,
@@ -66,6 +66,7 @@ def infer(
         width=width,
         height=height,
         generator=generator,
+        cross_attention_kwargs={"scale": lora_scale},
     ).images[0]
 
     return image, seed
@@ -198,6 +199,7 @@ with gr.Blocks(css=css) as demo:
             seed,
             guidance_scale,
             num_inference_steps,
+            lora_scale,
         ],
         outputs=[result, seed],
     )