Skip to content

Commit 6824031

Browse files
authored
[model] Support minimax 2.7 (#9079)
1 parent b6f20f6 commit 6824031

5 files changed

Lines changed: 12 additions & 0 deletions

File tree

docs/source/Instruction/Supported-models-and-datasets.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -573,6 +573,7 @@
573573
|[MiniMax/MiniMax-M2](https://modelscope.cn/models/MiniMax/MiniMax-M2)|minimax_m2|minimax_m2|transformers==4.57.1|✔|-|[MiniMaxAI/MiniMax-M2](https://huggingface.co/MiniMaxAI/MiniMax-M2)|
574574
|[MiniMax/MiniMax-M2.1](https://modelscope.cn/models/MiniMax/MiniMax-M2.1)|minimax_m2|minimax_m2_1|transformers==4.57.1|✔|-|[MiniMaxAI/MiniMax-M2.1](https://huggingface.co/MiniMaxAI/MiniMax-M2.1)|
575575
|[MiniMax/MiniMax-M2.5](https://modelscope.cn/models/MiniMax/MiniMax-M2.5)|minimax_m2|minimax_m2_5|transformers==4.57.1|✔|-|[MiniMaxAI/MiniMax-M2.5](https://huggingface.co/MiniMaxAI/MiniMax-M2.5)|
576+
|[MiniMax/MiniMax-M2.7](https://modelscope.cn/models/MiniMax/MiniMax-M2.7)|minimax_m2|minimax_m2_7|transformers==4.57.1|✔|-|[MiniMaxAI/MiniMax-M2.7](https://huggingface.co/MiniMaxAI/MiniMax-M2.7)|
576577
|[AI-ModelScope/gemma-2b-it](https://modelscope.cn/models/AI-ModelScope/gemma-2b-it)|gemma|gemma|transformers>=4.38|✘|-|[google/gemma-2b-it](https://huggingface.co/google/gemma-2b-it)|
577578
|[AI-ModelScope/gemma-2b](https://modelscope.cn/models/AI-ModelScope/gemma-2b)|gemma|gemma|transformers>=4.38|✘|-|[google/gemma-2b](https://huggingface.co/google/gemma-2b)|
578579
|[AI-ModelScope/gemma-7b](https://modelscope.cn/models/AI-ModelScope/gemma-7b)|gemma|gemma|transformers>=4.38|✘|-|[google/gemma-7b](https://huggingface.co/google/gemma-7b)|

docs/source_en/Instruction/Supported-models-and-datasets.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -574,6 +574,7 @@ The table below introduces the models integrated with ms-swift:
574574
|[MiniMax/MiniMax-M2](https://modelscope.cn/models/MiniMax/MiniMax-M2)|minimax_m2|minimax_m2|transformers==4.57.1|✔|-|[MiniMaxAI/MiniMax-M2](https://huggingface.co/MiniMaxAI/MiniMax-M2)|
575575
|[MiniMax/MiniMax-M2.1](https://modelscope.cn/models/MiniMax/MiniMax-M2.1)|minimax_m2|minimax_m2_1|transformers==4.57.1|✔|-|[MiniMaxAI/MiniMax-M2.1](https://huggingface.co/MiniMaxAI/MiniMax-M2.1)|
576576
|[MiniMax/MiniMax-M2.5](https://modelscope.cn/models/MiniMax/MiniMax-M2.5)|minimax_m2|minimax_m2_5|transformers==4.57.1|✔|-|[MiniMaxAI/MiniMax-M2.5](https://huggingface.co/MiniMaxAI/MiniMax-M2.5)|
577+
|[MiniMax/MiniMax-M2.7](https://modelscope.cn/models/MiniMax/MiniMax-M2.7)|minimax_m2|minimax_m2_7|transformers==4.57.1|✔|-|[MiniMaxAI/MiniMax-M2.7](https://huggingface.co/MiniMaxAI/MiniMax-M2.7)|
577578
|[AI-ModelScope/gemma-2b-it](https://modelscope.cn/models/AI-ModelScope/gemma-2b-it)|gemma|gemma|transformers>=4.38|✘|-|[google/gemma-2b-it](https://huggingface.co/google/gemma-2b-it)|
578579
|[AI-ModelScope/gemma-2b](https://modelscope.cn/models/AI-ModelScope/gemma-2b)|gemma|gemma|transformers>=4.38|✘|-|[google/gemma-2b](https://huggingface.co/google/gemma-2b)|
579580
|[AI-ModelScope/gemma-7b](https://modelscope.cn/models/AI-ModelScope/gemma-7b)|gemma|gemma|transformers>=4.38|✘|-|[google/gemma-7b](https://huggingface.co/google/gemma-7b)|

swift/model/models/minimax.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -158,6 +158,9 @@ def get_model(self, model_dir: str, config, processor, model_kwargs) -> PreTrain
158158
ModelGroup([
159159
Model('MiniMax/MiniMax-M2.5', 'MiniMaxAI/MiniMax-M2.5'),
160160
], TemplateType.minimax_m2_5),
161+
ModelGroup([
162+
Model('MiniMax/MiniMax-M2.7', 'MiniMaxAI/MiniMax-M2.7'),
163+
], TemplateType.minimax_m2_7),
161164
],
162165
requires=['transformers==4.57.1'],
163166
architectures=['MiniMaxM2ForCausalLM']))

swift/template/constant.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@ class LLMTemplateType:
4040
minimax_m2 = 'minimax_m2'
4141
minimax_m2_1 = 'minimax_m2_1'
4242
minimax_m2_5 = 'minimax_m2_5'
43+
minimax_m2_7 = 'minimax_m2_7'
4344
minimax_vl = 'minimax_vl'
4445

4546
numina = 'numina'

swift/template/templates/minimax.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -149,3 +149,9 @@ class MinimaxM2TemplateMeta(TemplateMeta):
149149
LLMTemplateType.minimax_m2_5,
150150
default_system='You are a helpful assistant. Your name is MiniMax-M2.5 and is built by MiniMax.',
151151
))
152+
153+
register_template(
154+
MinimaxM2TemplateMeta(
155+
LLMTemplateType.minimax_m2_7,
156+
default_system='You are a helpful assistant. Your name is MiniMax-M2.7 and is built by MiniMax.',
157+
))

0 commit comments

Comments
 (0)