Spaces:
Paused
Paused
Update app.py
Browse files
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")
|