Update app.py

app.py

@@ -5,6 +5,7 @@ import random
 # import spaces #[uncomment to use ZeroGPU]
 from diffusers import DiffusionPipeline, StableDiffusionPipeline
 import torch
+from peft import PeftModel
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
@@ -50,8 +51,19 @@ def get_lora_pipeline(base_model_id, lora_model_id, lora_scale=0.8):
             requires_safety_checker=False
         )
 
-        # адаптер
-        pipe.load_lora_weights(lora_model_id)
+        pipe.unet = PeftModel.from_pretrained(
+            pipe.unet, 
+            f"{lora_model_id}/unet",
+            adapter_name="default"
+        )
+        
+        pipe.text_encoder = PeftModel.from_pretrained(
+            pipe.text_encoder,
+            f"{lora_model_id}/text_encoder", 
+            adapter_name="default"
+        )
+        
+        pipe = pipe.to(device)
         
         pipe = pipe.to(device)
         lora_pipelines[cache_key] = pipe