Update app.py

app.py

@@ -41,7 +41,8 @@ def infer(
 
     if model == "Ramzes":
         pipe = StableDiffusionPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5", torch_dtype=torch_dtype)
-        pipe.unet = PeftModel.from_pretrained(pipe.unet, "Bordoglor/Ramzes_sd_v1.5/unet")
+        # pipe.unet = PeftModel.from_pretrained(pipe.unet, "Bordoglor/Ramzes_sd_v1.5/unet")
+        pipe.load_lora_weights("Bordoglor/Ramzes_sd_v1.5", subfolder="unet")
         pipe.text_encoder = PeftModel.from_pretrained(pipe.text_encoder, "Bordoglor/Ramzes_sd_v1.5/text_encoder")
     else:     
         pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch_dtype)