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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -45,8 +45,7 @@ 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.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")
@@ -97,7 +96,7 @@ with gr.Blocks() as demo:
97
  with gr.Column():
98
  image_in = gr.Image(label="Image Input", type="filepath")
99
  prompt = gr.Textbox(label="Prompt")
100
- orbit_type = gr.Radio(label="Orbit type", choices=["Left", "Up"], value="Left")
101
  submit_btn = gr.Button("Submit")
102
  with gr.Column():
103
  video_out = gr.Video(label="Video output")
 
45
  weight_name = "orbit_up_lora_weights.safetensors"
46
  adapter_name = "orbit_up_lora_weights"
47
  lora_rank = 256
48
+
 
49
  pipe.load_lora_weights(lora_path, weight_name=weight_name, adapter_name=adapter_name)
50
  pipe.fuse_lora(lora_scale=1 / lora_rank)
51
  pipe.to("cuda")
 
96
  with gr.Column():
97
  image_in = gr.Image(label="Image Input", type="filepath")
98
  prompt = gr.Textbox(label="Prompt")
99
+ orbit_type = gr.Radio(label="Orbit type", choices=["Left", "Up"], value="Left", interactive=False)
100
  submit_btn = gr.Button("Submit")
101
  with gr.Column():
102
  video_out = gr.Video(label="Video output")