mirror of
https://github.com/Wan-Video/Wan2.1.git
synced 2025-11-04 06:15:17 +00:00
66 lines
1.8 KiB
Python
66 lines
1.8 KiB
Python
# Copyright 2024-2025 The Alibaba Wan Team Authors. All rights reserved.
|
|
import copy
|
|
import os
|
|
|
|
os.environ['TOKENIZERS_PARALLELISM'] = 'false'
|
|
|
|
from .wan_i2v_14B import i2v_14B
|
|
from .wan_t2v_1_3B import t2v_1_3B
|
|
from .wan_t2v_14B import t2v_14B
|
|
|
|
# the config of t2i_14B is the same as t2v_14B
|
|
t2i_14B = copy.deepcopy(t2v_14B)
|
|
t2i_14B.__name__ = 'Config: Wan T2I 14B'
|
|
|
|
# the config of flf2v_14B is the same as i2v_14B
|
|
flf2v_14B = copy.deepcopy(i2v_14B)
|
|
flf2v_14B.__name__ = 'Config: Wan FLF2V 14B'
|
|
flf2v_14B.sample_neg_prompt = "镜头切换，" + flf2v_14B.sample_neg_prompt
|
|
|
|
WAN_CONFIGS = {
|
|
't2v-14B': t2v_14B,
|
|
't2v-1.3B': t2v_1_3B,
|
|
'i2v-14B': i2v_14B,
|
|
't2i-14B': t2i_14B,
|
|
'flf2v-14B': flf2v_14B,
|
|
'vace-1.3B': t2v_1_3B,
|
|
'vace-14B': t2v_14B,
|
|
}
|
|
|
|
SIZE_CONFIGS = {
|
|
'720*1280': (720, 1280),
|
|
'1280*720': (1280, 720),
|
|
'480*832': (480, 832),
|
|
'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),
|
|
}
|
|
|
|
MAX_AREA_CONFIGS = {
|
|
'720*1280': 720 * 1280,
|
|
'1280*720': 1280 * 720,
|
|
'480*832': 480 * 832,
|
|
'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', '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'),
|
|
'vace-14B': ('720*1280', '1280*720', '480*832', '832*480')
|
|
}
|