diff --git a/python/sglang/srt/models/commandr.py b/python/sglang/srt/models/commandr.py index 6d2e6d2bb..e4b291b66 100644 --- a/python/sglang/srt/models/commandr.py +++ b/python/sglang/srt/models/commandr.py @@ -386,4 +386,8 @@ class CohereForCausalLM(nn.Module): loaded_params.add(name) -EntryClass = CohereForCausalLM +class Cohere2ForCausalLM(CohereForCausalLM): + pass + + +EntryClass = [CohereForCausalLM, Cohere2ForCausalLM]