From 605db881d03cf0fac4b1bba604d9e6e5c18a2438 Mon Sep 17 00:00:00 2001 From: Kalila <69767640+digisomni@users.noreply.github.com> Date: Sat, 25 Nov 2023 22:48:06 +0800 Subject: [PATCH] Fix typos. --- fastchat/model/model_adapter.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/fastchat/model/model_adapter.py b/fastchat/model/model_adapter.py index fc8d1d98d..ba54f7db4 100644 --- a/fastchat/model/model_adapter.py +++ b/fastchat/model/model_adapter.py @@ -1893,7 +1893,7 @@ def get_default_conv_template(self, model_path: str) -> Conversation: return get_conv_template("aquila-chat") -class Lamma2ChineseAdapter(BaseModelAdapter): +class LLaMA2ChineseAdapter(BaseModelAdapter): """The model adapter for FlagAlpha/LLama2-Chinese sft""" def match(self, model_path: str): @@ -1918,7 +1918,7 @@ def get_default_conv_template(self, model_path: str) -> Conversation: return get_conv_template("llama2-chinese") -class Lamma2ChineseAlpacaAdapter(BaseModelAdapter): +class LLaMA2ChineseAlpacaAdapter(BaseModelAdapter): """The model adapter for ymcui/Chinese-LLaMA-Alpaca sft""" def match(self, model_path: str): @@ -2221,8 +2221,8 @@ def get_default_conv_template(self, model_path: str) -> Conversation: register_model_adapter(AquilaChatAdapter) register_model_adapter(BGEAdapter) register_model_adapter(E5Adapter) -register_model_adapter(Lamma2ChineseAdapter) -register_model_adapter(Lamma2ChineseAlpacaAdapter) +register_model_adapter(LLaMA2ChineseAdapter) +register_model_adapter(LLaMA2ChineseAlpacaAdapter) register_model_adapter(VigogneAdapter) register_model_adapter(OpenLLaMaOpenInstructAdapter) register_model_adapter(ReaLMAdapter)