Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
b4543c8f
Unverified
Commit
b4543c8f
authored
Apr 08, 2024
by
ywfang
Committed by
GitHub
Apr 08, 2024
Browse files
[Model] add minicpm (#3893)
parent
0ce0539d
Changes
4
Expand all
Show whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
543 additions
and
0 deletions
+543
-0
README.md
README.md
+1
-0
docs/source/models/supported_models.rst
docs/source/models/supported_models.rst
+4
-0
vllm/model_executor/models/__init__.py
vllm/model_executor/models/__init__.py
+1
-0
vllm/model_executor/models/minicpm.py
vllm/model_executor/models/minicpm.py
+537
-0
No files found.
README.md
View file @
b4543c8f
...
...
@@ -70,6 +70,7 @@ vLLM seamlessly supports many Hugging Face models, including the following archi
-
InternLM2 (
`internlm/internlm2-7b`
,
`internlm/internlm2-chat-7b`
, etc.)
-
Jais (
`core42/jais-13b`
,
`core42/jais-13b-chat`
,
`core42/jais-30b-v3`
,
`core42/jais-30b-chat-v3`
, etc.)
-
LLaMA & LLaMA-2 (
`meta-llama/Llama-2-70b-hf`
,
`lmsys/vicuna-13b-v1.3`
,
`young-geng/koala`
,
`openlm-research/open_llama_13b`
, etc.)
-
MiniCPM (
`openbmb/MiniCPM-2B-sft-bf16`
,
`openbmb/MiniCPM-2B-dpo-bf16`
, etc.)
-
Mistral (
`mistralai/Mistral-7B-v0.1`
,
`mistralai/Mistral-7B-Instruct-v0.1`
, etc.)
-
Mixtral (
`mistralai/Mixtral-8x7B-v0.1`
,
`mistralai/Mixtral-8x7B-Instruct-v0.1`
, etc.)
-
MPT (
`mosaicml/mpt-7b`
,
`mosaicml/mpt-30b`
, etc.)
...
...
docs/source/models/supported_models.rst
View file @
b4543c8f
...
...
@@ -83,6 +83,10 @@ Alongside each architecture, we include some popular models that use it.
- LLaMA, LLaMA-2, Vicuna, Alpaca, Yi
- :code:`meta-llama/Llama-2-13b-hf`, :code:`meta-llama/Llama-2-70b-hf`, :code:`openlm-research/open_llama_13b`, :code:`lmsys/vicuna-13b-v1.3`, :code:`01-ai/Yi-6B`, :code:`01-ai/Yi-34B`, etc.
- ✅︎
* - :code:`MiniCPMForCausalLM`
- MiniCPM
- :code:`openbmb/MiniCPM-2B-sft-bf16`, :code:`openbmb/MiniCPM-2B-dpo-bf16`, etc.
-
* - :code:`MistralForCausalLM`
- Mistral, Mistral-Instruct
- :code:`mistralai/Mistral-7B-v0.1`, :code:`mistralai/Mistral-7B-Instruct-v0.1`, etc.
...
...
vllm/model_executor/models/__init__.py
View file @
b4543c8f
...
...
@@ -41,6 +41,7 @@ _MODELS = {
# transformers's mpt class has lower case
"MptForCausalLM"
:
(
"mpt"
,
"MPTForCausalLM"
),
"MPTForCausalLM"
:
(
"mpt"
,
"MPTForCausalLM"
),
"MiniCPMForCausalLM"
:
(
"minicpm"
,
"MiniCPMForCausalLM"
),
"OLMoForCausalLM"
:
(
"olmo"
,
"OLMoForCausalLM"
),
"OPTForCausalLM"
:
(
"opt"
,
"OPTForCausalLM"
),
"OrionForCausalLM"
:
(
"orion"
,
"OrionForCausalLM"
),
...
...
vllm/model_executor/models/minicpm.py
0 → 100644
View file @
b4543c8f
This diff is collapsed.
Click to expand it.
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment