XiangpengYang commited on
Commit
c7eaeb0
·
1 Parent(s): e6a0a9e

wo acceleration lora, 4 step

Browse files
Files changed (2) hide show
  1. app.py +1 -21
  2. 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=True)
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(