linoyts HF Staff commited on
Commit
19e63cb
·
verified ·
1 Parent(s): dfce675

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -367,8 +367,9 @@ def generate(prompt,
367
  pipe.load_lora_weights(
368
  lora_repo_id,
369
  weight_name=lora_weights_name,
 
370
  )
371
- pipe.set_adapters(["active_lora"], adapter_weights=[lora_scale_value])
372
  pipe.fuse_lora(adapter_names=["active_lora"],lora_scale=lora_scale_value)
373
  pipe.unload_lora_weights()
374
  print(f"LoRA loaded into main pipe with scale {lora_scale_value}")
 
367
  pipe.load_lora_weights(
368
  lora_repo_id,
369
  weight_name=lora_weights_name,
370
+ adapter_name="active_lora"
371
  )
372
+ #pipe.set_adapters(["active_lora"], adapter_weights=[lora_scale_value])
373
  pipe.fuse_lora(adapter_names=["active_lora"],lora_scale=lora_scale_value)
374
  pipe.unload_lora_weights()
375
  print(f"LoRA loaded into main pipe with scale {lora_scale_value}")