Merge 66e5815491 into 7c81b2f27d

start_master.sh

@@ -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 "$@"

start_rank1.sh

@@ -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 "$@"

wan/configs/__init__.py

@@ -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'),