Update app.py

app.py

@@ -11,6 +11,15 @@ def get_lora_sd_pipeline(
     dtype=torch.float16, 
     adapter_name="default"
     ):
+
+    current_dir = os.getcwd()
+    print(f"Текущая рабочая директория: {current_dir}")
+
+    contents = os.listdir('.')
+    print("Содержимое директории:")
+    for item in contents:
+        print(item)
+
     unet_sub_dir = os.path.join(ckpt_dir, "unet")
     text_encoder_sub_dir = os.path.join(ckpt_dir, "text_encoder")
     
@@ -25,6 +34,7 @@ def get_lora_sd_pipeline(
     before_params = pipe.unet.parameters()
     pipe.unet = PeftModel.from_pretrained(pipe.unet, unet_sub_dir, adapter_name=adapter_name)
     pipe.fuse_lora(lora_scale=0.5)
+    pipe.unet.set_adapters_weight(lora_scale=0.5)
     after_params = pipe.unet.parameters()
     print("Parameters changed:", any(torch.any(b != a) for b, a in zip(before_params, after_params)))