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)