mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-08-25 15:00:07 +08:00
40 lines
911 B
Python
40 lines
911 B
Python
# Copyright 2024-2025 The Alibaba Wan Team Authors. All rights reserved.
|
|
import copy
|
|
import os
|
|
|
|
os.environ['TOKENIZERS_PARALLELISM'] = 'false'
|
|
|
|
from .wan_i2v_A14B import i2v_A14B
|
|
from .wan_t2v_A14B import t2v_A14B
|
|
from .wan_ti2v_5B import ti2v_5B
|
|
|
|
WAN_CONFIGS = {
|
|
't2v-A14B': t2v_A14B,
|
|
'i2v-A14B': i2v_A14B,
|
|
'ti2v-5B': ti2v_5B,
|
|
}
|
|
|
|
SIZE_CONFIGS = {
|
|
'720*1280': (720, 1280),
|
|
'1280*720': (1280, 720),
|
|
'480*832': (480, 832),
|
|
'832*480': (832, 480),
|
|
'704*1280': (704, 1280),
|
|
'1280*704': (1280, 704)
|
|
}
|
|
|
|
MAX_AREA_CONFIGS = {
|
|
'720*1280': 720 * 1280,
|
|
'1280*720': 1280 * 720,
|
|
'480*832': 480 * 832,
|
|
'832*480': 832 * 480,
|
|
'704*1280': 704 * 1280,
|
|
'1280*704': 1280 * 704,
|
|
}
|
|
|
|
SUPPORTED_SIZES = {
|
|
't2v-A14B': ('720*1280', '1280*720', '480*832', '832*480'),
|
|
'i2v-A14B': ('720*1280', '1280*720', '480*832', '832*480'),
|
|
'ti2v-5B': ('704*1280', '1280*704'),
|
|
}
|