Upload app.py

app.py

@@ -40,7 +40,7 @@ def load_pipeline(model_id: str):
         base_model = "CompVis/stable-diffusion-v1-4"
         pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch_dtype)
         # Load the LoRA weights into the U-Net.
-        pipe.unet = PeftModel.from_pretrained(pipe.unet, "unet")
+        pipe.unet = PeftModel.from_pretrained(pipe.unet, model_id, torch_dtype=torch_dtype)
 
     else:
         pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch_dtype)