jychen0828 commited on
Commit
d97b2e6
·
verified ·
1 Parent(s): e8d1004

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -53,6 +53,7 @@ def process_image(prompt,acc,height, width, steps, scales, seed):
53
  if loaded_acc != acc:
54
  #pipe.load_lora_weights(ACC_lora[acc], adapter_name=acc)
55
  pipe.set_adapters([acc], adapter_weights=[1.0])
 
56
  print(pipe.get_active_adapters())
57
  loaded_acc = acc
58
  with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16), timer("inference"):
 
53
  if loaded_acc != acc:
54
  #pipe.load_lora_weights(ACC_lora[acc], adapter_name=acc)
55
  pipe.set_adapters([acc], adapter_weights=[1.0])
56
+ pipe.fuse_lora(lora_scale=0.125)
57
  print(pipe.get_active_adapters())
58
  loaded_acc = acc
59
  with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16), timer("inference"):