├── .gitignore ├── LICENSE.md ├── README.md ├── configs ├── default_config.yaml ├── inference │ └── wan22.yaml ├── self_forcing_14b_dmd.yaml ├── self_forcing_14b_i2v_dmd.yaml ├── self_forcing_dmd.yaml ├── self_forcing_sid.yaml └── self_forcing_wan22_dmd.yaml ├── convert_checkpoint.py ├── data └── MagicData.csv ├── demo.py ├── demo_utils ├── constant.py ├── memory.py ├── taehv.py ├── utils.py ├── vae.py ├── vae_block3.py └── vae_torch2trt.py ├── demos ├── output_lightx2v_wan_t2v_t01.mp4 ├── output_lightx2v_wan_t2v_t02.mp4 ├── output_lightx2v_wan_t2v_t03.mp4 ├── output_lightx2v_wan_t2v_t04.mp4 ├── output_lightx2v_wan_t2v_t05.mp4 └── output_lightx2v_wan_t2v_t06.mp4 ├── examples ├── example.csv ├── images │ ├── cat.JPG │ ├── cup.jpg │ ├── guitar.jpeg │ ├── man.jpg │ ├── monster.png │ ├── robot.jpg │ └── tokyo.png └── videos │ ├── cat.mp4 │ ├── cup.mp4 │ ├── man.mp4 │ ├── monster.mp4 │ ├── robot.mp4 │ └── tokyo.mp4 ├── inference.py ├── model ├── __init__.py ├── base.py ├── causvid.py ├── diffusion.py ├── dmd.py ├── gan.py ├── ode_regression.py └── sid.py ├── pipeline ├── __init__.py ├── bidirectional_diffusion_inference.py ├── bidirectional_inference.py ├── bidirectional_training.py ├── causal_diffusion_inference.py ├── causal_inference.py ├── self_forcing_training.py └── wan22_fewstep_inference.py ├── prompts ├── MovieGenVideoBench.txt ├── MovieGenVideoBench_extended.txt └── vbench │ ├── all_dimension.txt │ └── all_dimension_longer.txt ├── requirements.txt ├── running_scripts ├── inference │ └── Wan2.2 │ │ └── i2v_fewstep.sh └── train │ └── Wan2.2 │ └── dmd.sh ├── scripts ├── compute_vae_latent.py ├── create_lmdb_14b_shards.py ├── create_lmdb_iterative.py └── generate_ode_pairs.py ├── setup.py ├── train.py ├── trainer ├── __init__.py ├── diffusion.py ├── distillation.py ├── gan.py ├── ode.py └── wan22_distillation.py ├── utils ├── dataset.py ├── distributed.py ├── lmdb.py ├── loss.py ├── misc.py ├── scheduler.py └── wan_wrapper.py ├── wan ├── README.md ├── __init__.py ├── configs │ ├── __init__.py │ ├── shared_config.py │ ├── wan_i2v_14B.py │ ├── wan_t2v_14B.py │ └── wan_t2v_1_3B.py ├── distributed │ ├── __init__.py │ ├── fsdp.py │ └── xdit_context_parallel.py ├── image2video.py ├── modules │ ├── __init__.py │ ├── attention.py │ ├── causal_model.py │ ├── clip.py │ ├── model.py │ ├── t5.py │ ├── tokenizers.py │ ├── vae.py │ └── xlm_roberta.py ├── text2video.py └── utils │ ├── __init__.py │ ├── fm_solvers.py │ ├── fm_solvers_unipc.py │ ├── prompt_extend.py │ ├── qwen_vl_utils.py │ └── utils.py ├── wan2.2_fewstep.py └── wan22 ├── __init__.py ├── configs ├── __init__.py ├── shared_config.py ├── wan_i2v_A14B.py ├── wan_t2v_A14B.py └── wan_ti2v_5B.py ├── distributed ├── __init__.py ├── fsdp.py ├── sequence_parallel.py ├── ulysses.py └── util.py ├── image2video.py ├── modules ├── __init__.py ├── attention.py ├── model.py ├── t5.py ├── tokenizers.py ├── vae2_1.py └── vae2_2.py ├── text2video.py ├── textimage2video.py └── utils ├── __init__.py ├── fm_solvers.py ├── fm_solvers_unipc.py ├── prompt_extend.py ├── qwen_vl_utils.py ├── system_prompt.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/README.md -------------------------------------------------------------------------------- /configs/default_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/configs/default_config.yaml -------------------------------------------------------------------------------- /configs/inference/wan22.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/configs/inference/wan22.yaml -------------------------------------------------------------------------------- /configs/self_forcing_14b_dmd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/configs/self_forcing_14b_dmd.yaml -------------------------------------------------------------------------------- /configs/self_forcing_14b_i2v_dmd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/configs/self_forcing_14b_i2v_dmd.yaml -------------------------------------------------------------------------------- /configs/self_forcing_dmd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/configs/self_forcing_dmd.yaml -------------------------------------------------------------------------------- /configs/self_forcing_sid.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/configs/self_forcing_sid.yaml -------------------------------------------------------------------------------- /configs/self_forcing_wan22_dmd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/configs/self_forcing_wan22_dmd.yaml -------------------------------------------------------------------------------- /convert_checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/convert_checkpoint.py -------------------------------------------------------------------------------- /data/MagicData.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/data/MagicData.csv -------------------------------------------------------------------------------- /demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/demo.py -------------------------------------------------------------------------------- /demo_utils/constant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/demo_utils/constant.py -------------------------------------------------------------------------------- /demo_utils/memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/demo_utils/memory.py -------------------------------------------------------------------------------- /demo_utils/taehv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/demo_utils/taehv.py -------------------------------------------------------------------------------- /demo_utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/demo_utils/utils.py -------------------------------------------------------------------------------- /demo_utils/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/demo_utils/vae.py -------------------------------------------------------------------------------- /demo_utils/vae_block3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/demo_utils/vae_block3.py -------------------------------------------------------------------------------- /demo_utils/vae_torch2trt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/demo_utils/vae_torch2trt.py -------------------------------------------------------------------------------- /demos/output_lightx2v_wan_t2v_t01.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/demos/output_lightx2v_wan_t2v_t01.mp4 -------------------------------------------------------------------------------- /demos/output_lightx2v_wan_t2v_t02.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/demos/output_lightx2v_wan_t2v_t02.mp4 -------------------------------------------------------------------------------- /demos/output_lightx2v_wan_t2v_t03.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/demos/output_lightx2v_wan_t2v_t03.mp4 -------------------------------------------------------------------------------- /demos/output_lightx2v_wan_t2v_t04.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/demos/output_lightx2v_wan_t2v_t04.mp4 -------------------------------------------------------------------------------- /demos/output_lightx2v_wan_t2v_t05.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/demos/output_lightx2v_wan_t2v_t05.mp4 -------------------------------------------------------------------------------- /demos/output_lightx2v_wan_t2v_t06.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/demos/output_lightx2v_wan_t2v_t06.mp4 -------------------------------------------------------------------------------- /examples/example.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/examples/example.csv -------------------------------------------------------------------------------- /examples/images/cat.JPG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/examples/images/cat.JPG -------------------------------------------------------------------------------- /examples/images/cup.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/examples/images/cup.jpg -------------------------------------------------------------------------------- /examples/images/guitar.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/examples/images/guitar.jpeg -------------------------------------------------------------------------------- /examples/images/man.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/examples/images/man.jpg -------------------------------------------------------------------------------- /examples/images/monster.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/examples/images/monster.png -------------------------------------------------------------------------------- /examples/images/robot.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/examples/images/robot.jpg -------------------------------------------------------------------------------- /examples/images/tokyo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/examples/images/tokyo.png -------------------------------------------------------------------------------- /examples/videos/cat.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/examples/videos/cat.mp4 -------------------------------------------------------------------------------- /examples/videos/cup.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/examples/videos/cup.mp4 -------------------------------------------------------------------------------- /examples/videos/man.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/examples/videos/man.mp4 -------------------------------------------------------------------------------- /examples/videos/monster.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/examples/videos/monster.mp4 -------------------------------------------------------------------------------- /examples/videos/robot.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/examples/videos/robot.mp4 -------------------------------------------------------------------------------- /examples/videos/tokyo.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/examples/videos/tokyo.mp4 -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/inference.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/model/__init__.py -------------------------------------------------------------------------------- /model/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/model/base.py -------------------------------------------------------------------------------- /model/causvid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/model/causvid.py -------------------------------------------------------------------------------- /model/diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/model/diffusion.py -------------------------------------------------------------------------------- /model/dmd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/model/dmd.py -------------------------------------------------------------------------------- /model/gan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/model/gan.py -------------------------------------------------------------------------------- /model/ode_regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/model/ode_regression.py -------------------------------------------------------------------------------- /model/sid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/model/sid.py -------------------------------------------------------------------------------- /pipeline/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/pipeline/__init__.py -------------------------------------------------------------------------------- /pipeline/bidirectional_diffusion_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/pipeline/bidirectional_diffusion_inference.py -------------------------------------------------------------------------------- /pipeline/bidirectional_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/pipeline/bidirectional_inference.py -------------------------------------------------------------------------------- /pipeline/bidirectional_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/pipeline/bidirectional_training.py -------------------------------------------------------------------------------- /pipeline/causal_diffusion_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/pipeline/causal_diffusion_inference.py -------------------------------------------------------------------------------- /pipeline/causal_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/pipeline/causal_inference.py -------------------------------------------------------------------------------- /pipeline/self_forcing_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/pipeline/self_forcing_training.py -------------------------------------------------------------------------------- /pipeline/wan22_fewstep_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/pipeline/wan22_fewstep_inference.py -------------------------------------------------------------------------------- /prompts/MovieGenVideoBench.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/prompts/MovieGenVideoBench.txt -------------------------------------------------------------------------------- /prompts/MovieGenVideoBench_extended.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/prompts/MovieGenVideoBench_extended.txt -------------------------------------------------------------------------------- /prompts/vbench/all_dimension.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/prompts/vbench/all_dimension.txt -------------------------------------------------------------------------------- /prompts/vbench/all_dimension_longer.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/prompts/vbench/all_dimension_longer.txt -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/requirements.txt -------------------------------------------------------------------------------- /running_scripts/inference/Wan2.2/i2v_fewstep.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/running_scripts/inference/Wan2.2/i2v_fewstep.sh -------------------------------------------------------------------------------- /running_scripts/train/Wan2.2/dmd.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/running_scripts/train/Wan2.2/dmd.sh -------------------------------------------------------------------------------- /scripts/compute_vae_latent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/scripts/compute_vae_latent.py -------------------------------------------------------------------------------- /scripts/create_lmdb_14b_shards.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/scripts/create_lmdb_14b_shards.py -------------------------------------------------------------------------------- /scripts/create_lmdb_iterative.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/scripts/create_lmdb_iterative.py -------------------------------------------------------------------------------- /scripts/generate_ode_pairs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/scripts/generate_ode_pairs.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/setup.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/train.py -------------------------------------------------------------------------------- /trainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/trainer/__init__.py -------------------------------------------------------------------------------- /trainer/diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/trainer/diffusion.py -------------------------------------------------------------------------------- /trainer/distillation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/trainer/distillation.py -------------------------------------------------------------------------------- /trainer/gan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/trainer/gan.py -------------------------------------------------------------------------------- /trainer/ode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/trainer/ode.py -------------------------------------------------------------------------------- /trainer/wan22_distillation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/trainer/wan22_distillation.py -------------------------------------------------------------------------------- /utils/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/utils/dataset.py -------------------------------------------------------------------------------- /utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/utils/distributed.py -------------------------------------------------------------------------------- /utils/lmdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/utils/lmdb.py -------------------------------------------------------------------------------- /utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/utils/loss.py -------------------------------------------------------------------------------- /utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/utils/misc.py -------------------------------------------------------------------------------- /utils/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/utils/scheduler.py -------------------------------------------------------------------------------- /utils/wan_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/utils/wan_wrapper.py -------------------------------------------------------------------------------- /wan/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan/README.md -------------------------------------------------------------------------------- /wan/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan/__init__.py -------------------------------------------------------------------------------- /wan/configs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan/configs/__init__.py -------------------------------------------------------------------------------- /wan/configs/shared_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan/configs/shared_config.py -------------------------------------------------------------------------------- /wan/configs/wan_i2v_14B.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan/configs/wan_i2v_14B.py -------------------------------------------------------------------------------- /wan/configs/wan_t2v_14B.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan/configs/wan_t2v_14B.py -------------------------------------------------------------------------------- /wan/configs/wan_t2v_1_3B.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan/configs/wan_t2v_1_3B.py -------------------------------------------------------------------------------- /wan/distributed/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /wan/distributed/fsdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan/distributed/fsdp.py -------------------------------------------------------------------------------- /wan/distributed/xdit_context_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan/distributed/xdit_context_parallel.py -------------------------------------------------------------------------------- /wan/image2video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan/image2video.py -------------------------------------------------------------------------------- /wan/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan/modules/__init__.py -------------------------------------------------------------------------------- /wan/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan/modules/attention.py -------------------------------------------------------------------------------- /wan/modules/causal_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan/modules/causal_model.py -------------------------------------------------------------------------------- /wan/modules/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan/modules/clip.py -------------------------------------------------------------------------------- /wan/modules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan/modules/model.py -------------------------------------------------------------------------------- /wan/modules/t5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan/modules/t5.py -------------------------------------------------------------------------------- /wan/modules/tokenizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan/modules/tokenizers.py -------------------------------------------------------------------------------- /wan/modules/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan/modules/vae.py -------------------------------------------------------------------------------- /wan/modules/xlm_roberta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan/modules/xlm_roberta.py -------------------------------------------------------------------------------- /wan/text2video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan/text2video.py -------------------------------------------------------------------------------- /wan/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan/utils/__init__.py -------------------------------------------------------------------------------- /wan/utils/fm_solvers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan/utils/fm_solvers.py -------------------------------------------------------------------------------- /wan/utils/fm_solvers_unipc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan/utils/fm_solvers_unipc.py -------------------------------------------------------------------------------- /wan/utils/prompt_extend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan/utils/prompt_extend.py -------------------------------------------------------------------------------- /wan/utils/qwen_vl_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan/utils/qwen_vl_utils.py -------------------------------------------------------------------------------- /wan/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan/utils/utils.py -------------------------------------------------------------------------------- /wan2.2_fewstep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan2.2_fewstep.py -------------------------------------------------------------------------------- /wan22/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan22/__init__.py -------------------------------------------------------------------------------- /wan22/configs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan22/configs/__init__.py -------------------------------------------------------------------------------- /wan22/configs/shared_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan22/configs/shared_config.py -------------------------------------------------------------------------------- /wan22/configs/wan_i2v_A14B.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan22/configs/wan_i2v_A14B.py -------------------------------------------------------------------------------- /wan22/configs/wan_t2v_A14B.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan22/configs/wan_t2v_A14B.py -------------------------------------------------------------------------------- /wan22/configs/wan_ti2v_5B.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan22/configs/wan_ti2v_5B.py -------------------------------------------------------------------------------- /wan22/distributed/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan22/distributed/__init__.py -------------------------------------------------------------------------------- /wan22/distributed/fsdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan22/distributed/fsdp.py -------------------------------------------------------------------------------- /wan22/distributed/sequence_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan22/distributed/sequence_parallel.py -------------------------------------------------------------------------------- /wan22/distributed/ulysses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan22/distributed/ulysses.py -------------------------------------------------------------------------------- /wan22/distributed/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan22/distributed/util.py -------------------------------------------------------------------------------- /wan22/image2video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan22/image2video.py -------------------------------------------------------------------------------- /wan22/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan22/modules/__init__.py -------------------------------------------------------------------------------- /wan22/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan22/modules/attention.py -------------------------------------------------------------------------------- /wan22/modules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan22/modules/model.py -------------------------------------------------------------------------------- /wan22/modules/t5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan22/modules/t5.py -------------------------------------------------------------------------------- /wan22/modules/tokenizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan22/modules/tokenizers.py -------------------------------------------------------------------------------- /wan22/modules/vae2_1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan22/modules/vae2_1.py -------------------------------------------------------------------------------- /wan22/modules/vae2_2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan22/modules/vae2_2.py -------------------------------------------------------------------------------- /wan22/text2video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan22/text2video.py -------------------------------------------------------------------------------- /wan22/textimage2video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan22/textimage2video.py -------------------------------------------------------------------------------- /wan22/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan22/utils/__init__.py -------------------------------------------------------------------------------- /wan22/utils/fm_solvers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan22/utils/fm_solvers.py -------------------------------------------------------------------------------- /wan22/utils/fm_solvers_unipc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan22/utils/fm_solvers_unipc.py -------------------------------------------------------------------------------- /wan22/utils/prompt_extend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan22/utils/prompt_extend.py -------------------------------------------------------------------------------- /wan22/utils/qwen_vl_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan22/utils/qwen_vl_utils.py -------------------------------------------------------------------------------- /wan22/utils/system_prompt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan22/utils/system_prompt.py -------------------------------------------------------------------------------- /wan22/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quanhaol/Wan2.2-TI2V-5B-Turbo/HEAD/wan22/utils/utils.py --------------------------------------------------------------------------------