Skip to content

Commit 9c3ce8b

Browse files
committed
added ltx2.3 model to common types and pyconfig
1 parent 4e17e19 commit 9c3ce8b

2 files changed

Lines changed: 3 additions & 2 deletions

File tree

src/maxdiffusion/common_types.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,7 @@
5252
WAN2_1 = "wan2.1"
5353
WAN2_2 = "wan2.2"
5454
LTX2_VIDEO = "ltx2_video"
55+
LTX2_3 = "ltx2.3"
5556

5657
WAN_MODEL = WAN2_1
5758

src/maxdiffusion/pyconfig.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,9 +27,9 @@
2727
from . import max_logging
2828
from . import max_utils
2929
from .models.wan.wan_utils import CAUSVID_TRANSFORMER_MODEL_NAME_OR_PATH, WAN_21_FUSION_X_MODEL_NAME_OR_PATH
30-
from maxdiffusion.common_types import LENGTH, KV_LENGTH, WAN2_1, WAN2_2, LTX2_VIDEO, RING_ATTENTION_AXIS_RULES, SEQUENCE_PARALLEL_AXIS_RULES
30+
from maxdiffusion.common_types import LENGTH, KV_LENGTH, WAN2_1, WAN2_2, LTX2_VIDEO, LTX2_3, RING_ATTENTION_AXIS_RULES, SEQUENCE_PARALLEL_AXIS_RULES
3131

32-
_ALLOWED_MODEL_NAMES = {WAN2_1, WAN2_2, LTX2_VIDEO}
32+
_ALLOWED_MODEL_NAMES = {WAN2_1, WAN2_2, LTX2_VIDEO, LTX2_3}
3333
_ALLOWED_TRAINING_MODEL_NAMES = {WAN2_1}
3434

3535

0 commit comments

Comments
 (0)