multimodalart HF staff commited on
Commit
accd1bc
1 Parent(s): 6560c53

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -31,10 +31,10 @@ def run(prompt, negative_prompt=None, guidance_scale=7.0, pag_scale=3.0, pag_lay
31
 
32
  if not prompt and not negative_prompt:
33
  guidance_scale = 0.0
34
- pipe.unload_lora_weights()
35
  pipe.unfuse_lora()
 
36
  if lora:
37
- pipe.load_lora_weights(lora, adapter_name="custom")
38
  pipe.fuse_lora(lora_scale=0.9)
39
  print(f"Seed before sending to generator for prompt: `{prompt}`", seed)
40
  generator = torch.Generator(device="cuda").manual_seed(seed)
 
31
 
32
  if not prompt and not negative_prompt:
33
  guidance_scale = 0.0
 
34
  pipe.unfuse_lora()
35
+ pipe.unload_lora_weights()
36
  if lora:
37
+ pipe.load_lora_weights(lora)
38
  pipe.fuse_lora(lora_scale=0.9)
39
  print(f"Seed before sending to generator for prompt: `{prompt}`", seed)
40
  generator = torch.Generator(device="cuda").manual_seed(seed)