File tree Expand file tree Collapse file tree 3 files changed +3
-3
lines changed Expand file tree Collapse file tree 3 files changed +3
-3
lines changed Original file line number Diff line number Diff line change 31
31
# not compatible with pip-compile.
32
32
"pfnet/plamo-2-1b" ,
33
33
"Zyphra/Zamba2-1.2B-instruct" ,
34
- "hmellor/bamba- tiny-random" ,
34
+ "hmellor/tiny-random-BambaForCausalLM " ,
35
35
]
36
36
37
37
# Avoid OOM
Original file line number Diff line number Diff line change @@ -124,7 +124,7 @@ def check_available_online(
124
124
"BaichuanForCausalLM" : _HfExamplesInfo ("baichuan-inc/Baichuan2-7B-chat" ,
125
125
trust_remote_code = True ),
126
126
"BambaForCausalLM" : _HfExamplesInfo ("ibm-ai-platform/Bamba-9B" ,
127
- extras = {"tiny" : "hmellor/bamba- tiny-random" }), # noqa: E501
127
+ extras = {"tiny" : "hmellor/tiny-random-BambaForCausalLM " }), # noqa: E501
128
128
"BloomForCausalLM" : _HfExamplesInfo ("bigscience/bloom-560m" ,
129
129
{"1b" : "bigscience/bloomz-1b1" }),
130
130
"ChatGLMModel" : _HfExamplesInfo ("THUDM/chatglm3-6b" ,
Original file line number Diff line number Diff line change 12
12
"openai/whisper-large-v3" , # transcription
13
13
"facebook/bart-large-cnn" , # encoder decoder
14
14
"mistralai/Mamba-Codestral-7B-v0.1" , # mamba
15
- "hmellor/bamba- tiny-random" , # hybrid
15
+ "hmellor/tiny-random-BambaForCausalLM " , # hybrid
16
16
"BAAI/bge-m3" , # embedding
17
17
]
18
18
You can’t perform that action at this time.
0 commit comments