fffiloni commited on
Commit
c7236d7
1 Parent(s): e65bce3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -45,6 +45,8 @@ def infer(image_path, prompt, orbit_type, progress=gr.Progress(track_tqdm=True))
45
  weight_name = "orbit_up_lora_weights.safetensors"
46
  adapter_name = "orbit_up_lora_weights"
47
  lora_rank = 256
 
 
48
  pipe.load_lora_weights(lora_path, weight_name=weight_name, adapter_name=adapter_name)
49
  pipe.fuse_lora(lora_scale=1 / lora_rank)
50
  pipe.to("cuda")
 
45
  weight_name = "orbit_up_lora_weights.safetensors"
46
  adapter_name = "orbit_up_lora_weights"
47
  lora_rank = 256
48
+ pipe.unfuse_lora()
49
+ pipe.unload_lora_weights()
50
  pipe.load_lora_weights(lora_path, weight_name=weight_name, adapter_name=adapter_name)
51
  pipe.fuse_lora(lora_scale=1 / lora_rank)
52
  pipe.to("cuda")