🐛 Bug Report: Error running wan2.1 — ui() missing required positional arguments

When running examples/wan2.1/app.py from the CogVideoX-Fun repo, I encountered the following error:

/worksapce/CogVideoX-Fun/examples/wan2.1/app.py
The cache for model files in Transformers v4.22.0 has been updated. Migrating your old cache. This is a one-time only operation. You can interrupt this and resume the migration later on by calling `transformers.utils.move_cache()`.            
0it [00:00, ?it/s]
/usr/local/lib/python3.10/dist-packages/xformers/ops/fmha/flash.py:211: FutureWarning: `torch.library.impl_abstract` was renamed to `torch.library.register_fake`. Please use that instead; we will remove `torch.library.impl_abstract` in a future version of PyTorch.
  @torch.library.impl_abstract("xformers_flash::flash_fwd")
/usr/local/lib/python3.10/dist-packages/xformers/ops/fmha/flash.py:344: FutureWarning: `torch.library.impl_abstract` was renamed to `torch.library.register_fake`. Please use that instead; we will remove `torch.library.impl_abstract` in a future version of PyTorch.
  @torch.library.impl_abstract("xformers_flash::flash_bwd")
/worksapce/CogVideoX-Fun/videox_fun/dist/wan_xfuser.py:32: FutureWarning: `torch.cuda.amp.autocast(args...)` is deprecated. Please use `torch.amp.autocast('cuda', args...)` instead.                                                              
  @amp.autocast(enabled=False)
/worksapce/CogVideoX-Fun/videox_fun/models/wan_transformer3d.py:212: FutureWarning: `torch.cuda.amp.autocast(args...)` is deprecated. Please use `torch.amp.autocast('cuda', args...)` instead.                                                    
  @amp.autocast(enabled=False)
/worksapce/CogVideoX-Fun/videox_fun/models/wan_transformer3d.py:223: FutureWarning: `torch.cuda.amp.autocast(args...)` is deprecated. Please use `torch.amp.autocast('cuda', args...)` instead.                                                    
  @amp.autocast(enabled=False)
/worksapce/CogVideoX-Fun/videox_fun/models/wan_transformer3d.py:282: FutureWarning: `torch.cuda.amp.autocast(args...)` is deprecated. Please use `torch.amp.autocast('cuda', args...)` instead.                                                    
  @amp.autocast(enabled=False)
Traceback (most recent call last):
  File "/worksapce/CogVideoX-Fun/examples/wan2.1/app.py", line 74, in <module>
    demo, controller = ui(GPU_memory_mode, flow_scheduler_dict, config_path, 1, 1, enable_teacache, teacache_threshold, num_skip_start_steps, teacache_offload, weight_dtype)                                                                      
TypeError: ui() missing 2 required positional arguments: 'riflex_k' and 'weight_dtype'

Steps to reproduce

I created a minimal Dockerfile based on your cogvideox_fun image:

FROM mybigpai-public-registry.cn-beijing.cr.aliyuncs.com/easycv/torch_cuda:cogvideox_fun

WORKDIR /worksapce

RUN git clone https://github.com/aigc-apps/CogVideoX-Fun.git

CMD ["/bin/bash", "sleep", "infinity"]

run command

 docker run --rm -it \
> --name videox \
> -e device=0,1,2,3 \
> --shm-size=100G \
> -v /data/Wan2.1/Wan2.1-I2V-14B-720P:/workspace/VideoX-Fun/models/Diffusion_Transformer/Wan2.1-I2V-14B-720P \
> --entrypoint python \
>   videox:v1 \
>   /worksapce/CogVideoX-Fun/examples/wan2.1/app.py

Start with examples/wan2.1/app.py, especially the ui() call at line 74, and compare it with the ui() definition to identify the missing arguments. Reproduce the command shown in the issue and confirm that the application starts without the positional-argument TypeError.