add ChatGLMForCausalLM

Signed-off-by: Alexandros Koumparoulis <[email protected]>
NVIDIA · Nov 20, 2024 · 433af57 · 433af57
1 parent da0ecee
commit 433af57
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/nemo/collections/llm/gpt/model/auto_model.py b/nemo/collections/llm/gpt/model/auto_model.py
@@ -35,6 +35,7 @@
     from nemo.collections.common.tokenizers.tokenizer_spec import TokenizerSpec
 
 HF_TO_MCORE_REGISTRY = {
+    'ChatGLMForCausalLM': llm.ChatGLMModel,
     'Phi3ForCausalLM': llm.Phi3Model,
     'MixtralForCausalLM': llm.MixtralModel,
     'MistralForCausalLM': llm.MistralModel,