Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
c7eaeb0
1
Parent(s):
e6a0a9e
wo acceleration lora, 4 step
Browse files- app.py +1 -21
- videox_fun/ui/ui.py +2 -1
app.py
CHANGED
|
@@ -128,13 +128,6 @@ class VideoCoF_Controller(Wan_Controller):
|
|
| 128 |
denoise_strength,
|
| 129 |
seed_textbox,
|
| 130 |
ref_image=None,
|
| 131 |
-
enable_teacache=None,
|
| 132 |
-
teacache_threshold=None,
|
| 133 |
-
num_skip_start_steps=None,
|
| 134 |
-
teacache_offload=None,
|
| 135 |
-
cfg_skip_ratio=None,
|
| 136 |
-
enable_riflex=None,
|
| 137 |
-
riflex_k=None,
|
| 138 |
# Custom args
|
| 139 |
source_frames_slider=33,
|
| 140 |
reasoning_frames_slider=4,
|
|
@@ -319,12 +312,6 @@ def ui(GPU_memory_mode, scheduler_dict, config_path, compile_dit, weight_dtype):
|
|
| 319 |
# Set default LoRA alpha to 1.0 (matching inference.py)
|
| 320 |
lora_alpha_slider.value = 1.0
|
| 321 |
|
| 322 |
-
with gr.Row():
|
| 323 |
-
# Disable teacache by default
|
| 324 |
-
enable_teacache, teacache_threshold, num_skip_start_steps, teacache_offload = create_teacache_params(False, 0.10, 5, False)
|
| 325 |
-
cfg_skip_ratio = create_cfg_skip_params(0)
|
| 326 |
-
enable_riflex, riflex_k = create_cfg_riflex_k(False, 6)
|
| 327 |
-
|
| 328 |
with gr.Column(variant="panel"):
|
| 329 |
prompt_textbox, negative_prompt_textbox = create_prompts(prompt="Remove the young man with short black hair wearing black shirt on the left.")
|
| 330 |
|
|
@@ -342,7 +329,7 @@ def ui(GPU_memory_mode, scheduler_dict, config_path, compile_dit, weight_dtype):
|
|
| 342 |
reasoning_frames_slider = gr.Slider(label="Reasoning Frames", minimum=1, maximum=20, value=4, step=1)
|
| 343 |
repeat_rope_checkbox = gr.Checkbox(label="Repeat RoPE", value=True)
|
| 344 |
# Add Acceleration Checkbox
|
| 345 |
-
enable_acceleration = gr.Checkbox(label="Enable 4-step Acceleration (FusionX LoRA)", value=
|
| 346 |
|
| 347 |
# Use custom height/width creation to hide/customize
|
| 348 |
resize_method, width_slider, height_slider, base_resolution = create_height_width_english(
|
|
@@ -408,13 +395,6 @@ def ui(GPU_memory_mode, scheduler_dict, config_path, compile_dit, weight_dtype):
|
|
| 408 |
denoise_strength,
|
| 409 |
seed_textbox,
|
| 410 |
ref_image,
|
| 411 |
-
enable_teacache,
|
| 412 |
-
teacache_threshold,
|
| 413 |
-
num_skip_start_steps,
|
| 414 |
-
teacache_offload,
|
| 415 |
-
cfg_skip_ratio,
|
| 416 |
-
enable_riflex,
|
| 417 |
-
riflex_k,
|
| 418 |
# New inputs
|
| 419 |
source_frames_slider,
|
| 420 |
reasoning_frames_slider,
|
|
|
|
| 128 |
denoise_strength,
|
| 129 |
seed_textbox,
|
| 130 |
ref_image=None,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 131 |
# Custom args
|
| 132 |
source_frames_slider=33,
|
| 133 |
reasoning_frames_slider=4,
|
|
|
|
| 312 |
# Set default LoRA alpha to 1.0 (matching inference.py)
|
| 313 |
lora_alpha_slider.value = 1.0
|
| 314 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 315 |
with gr.Column(variant="panel"):
|
| 316 |
prompt_textbox, negative_prompt_textbox = create_prompts(prompt="Remove the young man with short black hair wearing black shirt on the left.")
|
| 317 |
|
|
|
|
| 329 |
reasoning_frames_slider = gr.Slider(label="Reasoning Frames", minimum=1, maximum=20, value=4, step=1)
|
| 330 |
repeat_rope_checkbox = gr.Checkbox(label="Repeat RoPE", value=True)
|
| 331 |
# Add Acceleration Checkbox
|
| 332 |
+
enable_acceleration = gr.Checkbox(label="Enable 4-step Acceleration (FusionX LoRA)", value=False)
|
| 333 |
|
| 334 |
# Use custom height/width creation to hide/customize
|
| 335 |
resize_method, width_slider, height_slider, base_resolution = create_height_width_english(
|
|
|
|
| 395 |
denoise_strength,
|
| 396 |
seed_textbox,
|
| 397 |
ref_image,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 398 |
# New inputs
|
| 399 |
source_frames_slider,
|
| 400 |
reasoning_frames_slider,
|
videox_fun/ui/ui.py
CHANGED
|
@@ -41,7 +41,8 @@ def create_model_checkpoints(controller, visible, default_model="none"):
|
|
| 41 |
gr.Markdown(
|
| 42 |
"""
|
| 43 |
### Model checkpoints.
|
| 44 |
-
"""
|
|
|
|
| 45 |
)
|
| 46 |
with gr.Row(visible=visible):
|
| 47 |
diffusion_transformer_dropdown = gr.Dropdown(
|
|
|
|
| 41 |
gr.Markdown(
|
| 42 |
"""
|
| 43 |
### Model checkpoints.
|
| 44 |
+
""",
|
| 45 |
+
visible=visible
|
| 46 |
)
|
| 47 |
with gr.Row(visible=visible):
|
| 48 |
diffusion_transformer_dropdown = gr.Dropdown(
|