ALSv/self-forcing

1# Copyright 2024-2025 The Alibaba Wan Team Authors. All rights reserved.2from .wan_t2v_14B import t2v_14B3from .wan_t2v_1_3B import t2v_1_3B4from .wan_i2v_14B import i2v_14B5import copy6import os7 8os.environ['TOKENIZERS_PARALLELISM'] = 'false'9 10 11# the config of t2i_14B is the same as t2v_14B12t2i_14B = copy.deepcopy(t2v_14B)13t2i_14B.__name__ = 'Config: Wan T2I 14B'14 15WAN_CONFIGS = {16    't2v-14B': t2v_14B,17    't2v-1.3B': t2v_1_3B,18    'i2v-14B': i2v_14B,19    't2i-14B': t2i_14B,20}21 22SIZE_CONFIGS = {23    '720*1280': (720, 1280),24    '1280*720': (1280, 720),25    '480*832': (480, 832),26    '832*480': (832, 480),27    '1024*1024': (1024, 1024),28}29 30MAX_AREA_CONFIGS = {31    '720*1280': 720 * 1280,32    '1280*720': 1280 * 720,33    '480*832': 480 * 832,34    '832*480': 832 * 480,35}36 37SUPPORTED_SIZES = {38    't2v-14B': ('720*1280', '1280*720', '480*832', '832*480'),39    't2v-1.3B': ('480*832', '832*480'),40    'i2v-14B': ('720*1280', '1280*720', '480*832', '832*480'),41    't2i-14B': tuple(SIZE_CONFIGS.keys()),42}43