fix custom (modelscope#1028)

tastelikefeet · May 30, 2024 · f3b5d42 · f3b5d42
1 parent 53ec6d7
commit f3b5d42
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 9 deletions.
diff --git a/docs/source/LLM/自定义与拓展.md b/docs/source/LLM/自定义与拓展.md
@@ -229,13 +229,13 @@ class CustomTemplateType:
 
 
 @register_model(CustomModelType.tigerbot_7b,
-                'TigerResearch/tigerbot-7b-base-v3', LoRATM.llama2,
+                'TigerResearch/tigerbot-7b-base-v3', LoRATM.llama,
                 TemplateType.default_generation)
 @register_model(CustomModelType.tigerbot_13b,
-                'TigerResearch/tigerbot-13b-base-v2', LoRATM.llama2,
+                'TigerResearch/tigerbot-13b-base-v2', LoRATM.llama,
                 TemplateType.default_generation)
 @register_model(CustomModelType.tigerbot_13b_chat,
-                'TigerResearch/tigerbot-13b-chat-v4', LoRATM.llama2,
+                'TigerResearch/tigerbot-13b-chat-v4', LoRATM.llama,
                 CustomTemplateType.tigerbot)
 def get_tigerbot_model_tokenizer(model_dir: str,
                                  torch_dtype: Dtype,

diff --git a/docs/source_en/LLM/Customization.md b/docs/source_en/LLM/Customization.md
@@ -237,13 +237,13 @@ class CustomTemplateType:
 
 
 @register_model(CustomModelType.tigerbot_7b,
-                'TigerResearch/tigerbot-7b-base-v3', LoRATM.llama2,
+                'TigerResearch/tigerbot-7b-base-v3', LoRATM.llama,
                 TemplateType.default_generation)
 @register_model(CustomModelType.tigerbot_13b,
-                'TigerResearch/tigerbot-13b-base-v2', LoRATM.llama2,
+                'TigerResearch/tigerbot-13b-base-v2', LoRATM.llama,
                 TemplateType.default_generation)
 @register_model(CustomModelType.tigerbot_13b_chat,
-                'TigerResearch/tigerbot-13b-chat-v4', LoRATM.llama2,
+                'TigerResearch/tigerbot-13b-chat-v4', LoRATM.llama,
                 CustomTemplateType.tigerbot)
 def get_tigerbot_model_tokenizer(model_dir: str,
                                  torch_dtype: Dtype,

diff --git a/examples/pytorch/llm/custom.py b/examples/pytorch/llm/custom.py
@@ -28,11 +28,11 @@ class CustomDatasetName:
     stsb_en = 'stsb-en'
 
 
-@register_model(CustomModelType.tigerbot_7b, 'TigerResearch/tigerbot-7b-base-v3', LoRATM.llama2,
+@register_model(CustomModelType.tigerbot_7b, 'TigerResearch/tigerbot-7b-base-v3', LoRATM.llama,
                 TemplateType.default_generation)
-@register_model(CustomModelType.tigerbot_13b, 'TigerResearch/tigerbot-13b-base-v2', LoRATM.llama2,
+@register_model(CustomModelType.tigerbot_13b, 'TigerResearch/tigerbot-13b-base-v2', LoRATM.llama,
                 TemplateType.default_generation)
-@register_model(CustomModelType.tigerbot_13b_chat, 'TigerResearch/tigerbot-13b-chat-v4', LoRATM.llama2,
+@register_model(CustomModelType.tigerbot_13b_chat, 'TigerResearch/tigerbot-13b-chat-v4', LoRATM.llama,
                 CustomTemplateType.tigerbot)
 def get_tigerbot_model_tokenizer(model_dir: str,
                                  torch_dtype: Dtype,