Update app.py

app.py

@@ -72,7 +72,8 @@ def infer(
               'num_inference_steps': num_inference_steps,
               'width': width,
               'height': height,
-              'generator': generator
+              'generator': generator,
+              'cross_attention_kwargs': {"scale": lora_scale}
              }
 
     if controlnet_checkbox:
@@ -120,8 +121,8 @@ def infer(
     pipe.unet = PeftModel.from_pretrained(pipe.unet, unet_sub_dir)
     pipe.text_encoder = PeftModel.from_pretrained(pipe.text_encoder, text_encoder_sub_dir)
 
-    pipe.unet.add_weighted_adapter(['default'], [lora_scale], 'lora')
-    pipe.text_encoder.add_weighted_adapter(['default'], [lora_scale], 'lora')
+    # pipe.unet.add_weighted_adapter(['default'], [lora_scale], 'lora')
+    # pipe.text_encoder.add_weighted_adapter(['default'], [lora_scale], 'lora')
 
     # pipe.unet.load_state_dict({k: lora_scale*v for k, v in pipe.unet.state_dict().items()})
     # pipe.text_encoder.load_state_dict({k: lora_scale*v for k, v in pipe.text_encoder.state_dict().items()})