Skip to content

Commit fc088e6

Browse files
dmtri35nvxuanyuc
andauthored
[None][feat] Support Glm4MoeForCausalLM (#8256)
Signed-off-by: Tri Dao <[email protected]> Co-authored-by: Xuanyu Chen <[email protected]>
1 parent c3376fa commit fc088e6

File tree

8 files changed

+1010
-9
lines changed

8 files changed

+1010
-9
lines changed

tensorrt_llm/_torch/models/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
from .modeling_exaone4 import Exaone4ForCausalLM
88
from .modeling_gemma3 import Gemma3ForCausalLM
99
from .modeling_gemma3vl import Gemma3VLM
10+
from .modeling_glm import Glm4MoeForCausalLM
1011
from .modeling_gpt_oss import GptOssForCausalLM
1112
from .modeling_hunyuan_dense import HunYuanDenseV1ForCausalLM
1213
from .modeling_hunyuan_moe import HunYuanMoEV1ForCausalLM
@@ -70,6 +71,7 @@
7071
"Qwen3NextForCausalLM",
7172
"GptOssForCausalLM",
7273
"SeedOssForCausalLM",
74+
"Glm4MoeForCausalLM",
7375
]
7476

7577
if transformers.__version__ >= "4.45.1":

0 commit comments

Comments
 (0)