Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-12-20 14:12:04 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Compare commits

base: Zengtudor:46af9d35dfe2153a1647bb40602b23df8f641040
Branches Tags
Zengtudor:main
Zengtudor:dev/format
..
compare: Zengtudor:7a484a0350b31c783da635553e5dad5fa05ab2c2
Branches Tags
Zengtudor:main
Zengtudor:dev/format

4 Commits
46af9d35df ... 7a484a0350

Author SHA1 Message Date
error.d
7a484a0350
Merge 66e5815491 into 7c81b2f27d 2025-07-23 08:12:40 -07:00
ErrorDing
66e5815491 Change file permissions for start_master.sh and start_rank1.sh to make them executable. 2025-07-23 11:16:16 +08:00
ErrorDing
dc96cbb091 Expand supported resolutions in SIZE_CONFIGS and MAX_AREA_CONFIGS; update SUPPORTED_SIZES for i2v-14B to include new dimensions. 2025-07-23 11:14:55 +08:00
ErrorDing
a51658f197 Update script usage examples in start_master.sh and start_rank1.sh to include --sample_steps parameter 2025-07-23 10:37:07 +08:00
3 changed files with 11 additions and 3 deletions
Show Stats Expand all files Collapse all files

2
start_master.sh Normal file → Executable file
View File

@ -10,5 +10,5 @@ export LOCAL_RANK='0'
export NCCL_SOCKET_IFNAME='enp65s0'
# Run the generate.py script with specified parameters
# like: ./start_master.sh --dit_fsdp --task i2v-14B --size 480*640 --ckpt_dir ./Wan2.1-I2V-14B-480P --image examples/i2v_input.JPG --prompt "Summer beach vacation style, a white cat wearing sunglasses sits on a surfboard. The fluffy-furred feline gazes directly at the camera with a relaxed expression. Blurred beach scenery forms the background featuring crystal-clear waters, distant green hills, and a blue sky dotted with white clouds. The cat assumes a naturally relaxed posture, as if savoring the sea breeze and warm sunlight. A close-up shot highlights the feline's intricate details and the refreshing atmosphere of the seaside."
# like: ./start_master.sh --dit_fsdp --sample_steps 40--task i2v-14B --size 480*640 --ckpt_dir ./Wan2.1-I2V-14B-480P --image examples/i2v_input.JPG --prompt "Summer beach vacation style, a white cat wearing sunglasses sits on a surfboard. The fluffy-furred feline gazes directly at the camera with a relaxed expression. Blurred beach scenery forms the background featuring crystal-clear waters, distant green hills, and a blue sky dotted with white clouds. The cat assumes a naturally relaxed posture, as if savoring the sea breeze and warm sunlight. A close-up shot highlights the feline's intricate details and the refreshing atmosphere of the seaside."
python generate.py "$@"

2
start_rank1.sh Normal file → Executable file
View File

@ -10,5 +10,5 @@ export LOCAL_RANK='0'
export NCCL_SOCKET_IFNAME='enp33s0'
# Run the generate.py script with specified parameters
# like: ./start_rank1.sh --dit_fsdp --task i2v-14B --size 480*640 --ckpt_dir ./Wan2.1-I2V-14B-480P --image examples/i2v_input.JPG --prompt "Summer beach vacation style, a white cat wearing sunglasses sits on a surfboard. The fluffy-furred feline gazes directly at the camera with a relaxed expression. Blurred beach scenery forms the background featuring crystal-clear waters, distant green hills, and a blue sky dotted with white clouds. The cat assumes a naturally relaxed posture, as if savoring the sea breeze and warm sunlight. A close-up shot highlights the feline's intricate details and the refreshing atmosphere of the seaside."
# like: ./start_rank1.sh --dit_fsdp --sample_steps 40 --task i2v-14B --size 480*640 --ckpt_dir ./Wan2.1-I2V-14B-480P --image examples/i2v_input.JPG --prompt "Summer beach vacation style, a white cat wearing sunglasses sits on a surfboard. The fluffy-furred feline gazes directly at the camera with a relaxed expression. Blurred beach scenery forms the background featuring crystal-clear waters, distant green hills, and a blue sky dotted with white clouds. The cat assumes a naturally relaxed posture, as if savoring the sea breeze and warm sunlight. A close-up shot highlights the feline's intricate details and the refreshing atmosphere of the seaside."
python generate.py "$@"

10
wan/configs/__init__.py
View File

@ -34,6 +34,10 @@ SIZE_CONFIGS = {
'832*480': (832, 480),
'480*640': (480, 640),
'640*480': (640, 480),
'640*848': (640, 848),
'848*640': (848, 640),
'720*960': (720, 960),
'960*720': (960, 720),
'1024*1024': (1024, 1024),
}
@ -44,12 +48,16 @@ MAX_AREA_CONFIGS = {
'832*480': 832 * 480,
'480*640': 480 * 640,
'640*480': 640 * 480,
'640*848': 640 * 848,
'848*640': 848 * 640,
'720*960': 720 * 960,
'960*720': 960 * 720,
}
SUPPORTED_SIZES = {
't2v-14B': ('720*1280', '1280*720', '480*832', '832*480'),
't2v-1.3B': ('480*832', '832*480'),
'i2v-14B': ('720*1280', '1280*720', '480*832', '832*480', '480*640', '640*480'),
'i2v-14B': ('720*1280', '1280*720', '480*832', '832*480', '480*640', '640*480', '640*848', '848*640', '720*960', '960*720'),
'flf2v-14B': ('720*1280', '1280*720', '480*832', '832*480'),
't2i-14B': tuple(SIZE_CONFIGS.keys()),
'vace-1.3B': ('480*832', '832*480'),