EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

Enable Cohere2 Models (#3018)

Browse Source
This commit is contained in:
Hui Liu
2025-01-20 19:21:41 -08:00
committed by GitHub
parent b730aa6b9e
commit d2571dd5c7
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
python/sglang/srt/models/commandr.py
View File

@@ -386,4 +386,8 @@ class CohereForCausalLM(nn.Module):
loaded_params.add(name) loaded_params.add(name)
EntryClass = CohereForCausalLM class Cohere2ForCausalLM(CohereForCausalLM):
pass
EntryClass = [CohereForCausalLM, Cohere2ForCausalLM]