├── .gitignore ├── LICENSE ├── README.md ├── caption_videos_qwen.py ├── convert_lora_to_comfy.py ├── embeddings ├── default_empty.txt ├── default_empty_wan.safetensors ├── default_image_negative.txt ├── default_image_negative_wan.safetensors ├── default_video_negative.txt └── default_video_negative_wan.safetensors ├── launch_venv.bat ├── merge_last_n_checkpoints.py ├── requirements.txt ├── test_wan_control_lora.py ├── train_wan_lora.py ├── utils ├── dataset.py ├── depth_anything_v2 │ ├── dinov2.py │ ├── dinov2_layers │ │ ├── __init__.py │ │ ├── attention.py │ │ ├── block.py │ │ ├── drop_path.py │ │ ├── layer_scale.py │ │ ├── mlp.py │ │ ├── patch_embed.py │ │ └── swiglu_ffn.py │ ├── dpt.py │ └── util │ │ ├── blocks.py │ │ └── transform.py ├── dwt_loss.py └── temp_rng.py └── wan ├── __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 ├── 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 /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/README.md -------------------------------------------------------------------------------- /caption_videos_qwen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/caption_videos_qwen.py -------------------------------------------------------------------------------- /convert_lora_to_comfy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/convert_lora_to_comfy.py -------------------------------------------------------------------------------- /embeddings/default_empty.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /embeddings/default_empty_wan.safetensors: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/embeddings/default_empty_wan.safetensors -------------------------------------------------------------------------------- /embeddings/default_image_negative.txt: -------------------------------------------------------------------------------- 1 | watermark, 构图不佳, poor composition, 色彩艳丽, 模糊, 比例失调, 留白过多, low resolution -------------------------------------------------------------------------------- /embeddings/default_image_negative_wan.safetensors: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/embeddings/default_image_negative_wan.safetensors -------------------------------------------------------------------------------- /embeddings/default_video_negative.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/embeddings/default_video_negative.txt -------------------------------------------------------------------------------- /embeddings/default_video_negative_wan.safetensors: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/embeddings/default_video_negative_wan.safetensors -------------------------------------------------------------------------------- /launch_venv.bat: -------------------------------------------------------------------------------- 1 | @echo off 2 | call %~dp0.venv\Scripts\activate.bat 3 | cmd /K -------------------------------------------------------------------------------- /merge_last_n_checkpoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/merge_last_n_checkpoints.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/requirements.txt -------------------------------------------------------------------------------- /test_wan_control_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/test_wan_control_lora.py -------------------------------------------------------------------------------- /train_wan_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/train_wan_lora.py -------------------------------------------------------------------------------- /utils/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/utils/dataset.py -------------------------------------------------------------------------------- /utils/depth_anything_v2/dinov2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/utils/depth_anything_v2/dinov2.py -------------------------------------------------------------------------------- /utils/depth_anything_v2/dinov2_layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/utils/depth_anything_v2/dinov2_layers/__init__.py -------------------------------------------------------------------------------- /utils/depth_anything_v2/dinov2_layers/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/utils/depth_anything_v2/dinov2_layers/attention.py -------------------------------------------------------------------------------- /utils/depth_anything_v2/dinov2_layers/block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/utils/depth_anything_v2/dinov2_layers/block.py -------------------------------------------------------------------------------- /utils/depth_anything_v2/dinov2_layers/drop_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/utils/depth_anything_v2/dinov2_layers/drop_path.py -------------------------------------------------------------------------------- /utils/depth_anything_v2/dinov2_layers/layer_scale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/utils/depth_anything_v2/dinov2_layers/layer_scale.py -------------------------------------------------------------------------------- /utils/depth_anything_v2/dinov2_layers/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/utils/depth_anything_v2/dinov2_layers/mlp.py -------------------------------------------------------------------------------- /utils/depth_anything_v2/dinov2_layers/patch_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/utils/depth_anything_v2/dinov2_layers/patch_embed.py -------------------------------------------------------------------------------- /utils/depth_anything_v2/dinov2_layers/swiglu_ffn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/utils/depth_anything_v2/dinov2_layers/swiglu_ffn.py -------------------------------------------------------------------------------- /utils/depth_anything_v2/dpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/utils/depth_anything_v2/dpt.py -------------------------------------------------------------------------------- /utils/depth_anything_v2/util/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/utils/depth_anything_v2/util/blocks.py -------------------------------------------------------------------------------- /utils/depth_anything_v2/util/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/utils/depth_anything_v2/util/transform.py -------------------------------------------------------------------------------- /utils/dwt_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/utils/dwt_loss.py -------------------------------------------------------------------------------- /utils/temp_rng.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/utils/temp_rng.py -------------------------------------------------------------------------------- /wan/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/wan/__init__.py -------------------------------------------------------------------------------- /wan/configs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/wan/configs/__init__.py -------------------------------------------------------------------------------- /wan/configs/shared_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/wan/configs/shared_config.py -------------------------------------------------------------------------------- /wan/configs/wan_i2v_14B.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/wan/configs/wan_i2v_14B.py -------------------------------------------------------------------------------- /wan/configs/wan_t2v_14B.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/wan/configs/wan_t2v_14B.py -------------------------------------------------------------------------------- /wan/configs/wan_t2v_1_3B.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/wan/configs/wan_t2v_1_3B.py -------------------------------------------------------------------------------- /wan/distributed/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /wan/distributed/fsdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/wan/distributed/fsdp.py -------------------------------------------------------------------------------- /wan/distributed/xdit_context_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/wan/distributed/xdit_context_parallel.py -------------------------------------------------------------------------------- /wan/image2video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/wan/image2video.py -------------------------------------------------------------------------------- /wan/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/wan/modules/__init__.py -------------------------------------------------------------------------------- /wan/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/wan/modules/attention.py -------------------------------------------------------------------------------- /wan/modules/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/wan/modules/clip.py -------------------------------------------------------------------------------- /wan/modules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/wan/modules/model.py -------------------------------------------------------------------------------- /wan/modules/t5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/wan/modules/t5.py -------------------------------------------------------------------------------- /wan/modules/tokenizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/wan/modules/tokenizers.py -------------------------------------------------------------------------------- /wan/modules/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/wan/modules/vae.py -------------------------------------------------------------------------------- /wan/modules/xlm_roberta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/wan/modules/xlm_roberta.py -------------------------------------------------------------------------------- /wan/text2video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/wan/text2video.py -------------------------------------------------------------------------------- /wan/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/wan/utils/__init__.py -------------------------------------------------------------------------------- /wan/utils/fm_solvers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/wan/utils/fm_solvers.py -------------------------------------------------------------------------------- /wan/utils/fm_solvers_unipc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/wan/utils/fm_solvers_unipc.py -------------------------------------------------------------------------------- /wan/utils/prompt_extend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/wan/utils/prompt_extend.py -------------------------------------------------------------------------------- /wan/utils/qwen_vl_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/wan/utils/qwen_vl_utils.py -------------------------------------------------------------------------------- /wan/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/spacepxl/WanTraining/HEAD/wan/utils/utils.py --------------------------------------------------------------------------------