jbilcke-hf HF Staff commited on
Commit
868becf
·
verified ·
1 Parent(s): ea13288

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -120,7 +120,7 @@ def generate_video(prompt, height, width,
120
  return video_path, current_seed
121
 
122
  with gr.Blocks() as demo:
123
- gr.Markdown("# Fast 4 steps Wan 2.1 T2V (1.3B) with CausVid LoRA")
124
  gr.Markdown("[CausVid](https://github.com/tianweiy/CausVid) is a distilled version of Wan 2.1 to run faster in just 4-8 steps, [extracted as LoRA by Kijai](https://huggingface.co/Kijai/WanVideo_comfy/blob/main/Wan21_CausVid_bidirect2_T2V_1_3B_lora_rank32.safetensors) and is compatible with 🧨 diffusers")
125
  with gr.Row():
126
  with gr.Column():
@@ -134,7 +134,7 @@ with gr.Blocks() as demo:
134
  with gr.Row():
135
  height_input = gr.Slider(minimum=SLIDER_MIN_H, maximum=SLIDER_MAX_H, step=MOD_VALUE, value=DEFAULT_H_SLIDER_VALUE, label=f"Output Height (multiple of {MOD_VALUE})")
136
  width_input = gr.Slider(minimum=SLIDER_MIN_W, maximum=SLIDER_MAX_W, step=MOD_VALUE, value=DEFAULT_W_SLIDER_VALUE, label=f"Output Width (multiple of {MOD_VALUE})")
137
- steps_slider = gr.Slider(minimum=1, maximum=30, step=1, value=4, label="Inference Steps")
138
  guidance_scale_input = gr.Slider(minimum=0.0, maximum=20.0, step=0.5, value=1.0, label="Guidance Scale", visible=False)
139
 
140
  generate_button = gr.Button("Generate Video", variant="primary")
 
120
  return video_path, current_seed
121
 
122
  with gr.Blocks() as demo:
123
+ gr.Markdown("# Ultra Fast 4 steps Wan 2.1 T2V (1.3B) with CausVid LoRA")
124
  gr.Markdown("[CausVid](https://github.com/tianweiy/CausVid) is a distilled version of Wan 2.1 to run faster in just 4-8 steps, [extracted as LoRA by Kijai](https://huggingface.co/Kijai/WanVideo_comfy/blob/main/Wan21_CausVid_bidirect2_T2V_1_3B_lora_rank32.safetensors) and is compatible with 🧨 diffusers")
125
  with gr.Row():
126
  with gr.Column():
 
134
  with gr.Row():
135
  height_input = gr.Slider(minimum=SLIDER_MIN_H, maximum=SLIDER_MAX_H, step=MOD_VALUE, value=DEFAULT_H_SLIDER_VALUE, label=f"Output Height (multiple of {MOD_VALUE})")
136
  width_input = gr.Slider(minimum=SLIDER_MIN_W, maximum=SLIDER_MAX_W, step=MOD_VALUE, value=DEFAULT_W_SLIDER_VALUE, label=f"Output Width (multiple of {MOD_VALUE})")
137
+ steps_slider = gr.Slider(minimum=1, maximum=8, step=1, value=3, label="Inference Steps")
138
  guidance_scale_input = gr.Slider(minimum=0.0, maximum=20.0, step=0.5, value=1.0, label="Guidance Scale", visible=False)
139
 
140
  generate_button = gr.Button("Generate Video", variant="primary")