feat: support internlm 3 dense (#2888)

This commit is contained in:
Yineng Zhang
2025-01-14 19:23:26 +08:00
committed by GitHub
parent cc0485bef2
commit f5c6c66794

View File

@@ -570,4 +570,8 @@ class Phi3ForCausalLM(LlamaForCausalLM):
pass
EntryClass = [LlamaForCausalLM, Phi3ForCausalLM]
class InternLM3ForCausalLM(LlamaForCausalLM):
pass
EntryClass = [LlamaForCausalLM, Phi3ForCausalLM, InternLM3ForCausalLM]