Skip to content

Commit

Permalink
Update model_adapter.py (lm-sys#2895)
Browse files Browse the repository at this point in the history
  • Loading branch information
thavens authored and zhanghao.smooth committed Jan 26, 2024
1 parent 09ba8d5 commit 4d113fb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion fastchat/model/model_adapter.py
Original file line number Diff line number Diff line change
Expand Up @@ -2020,7 +2020,7 @@ class TinyLlamaAdapter(BaseModelAdapter):
"""The model adapter for TinyLlama (e.g. TinyLlama/TinyLlama-1.1B-Chat-v1.0)"""

def match(self, model_path: str):
return "TinyLlama" in model_path.lower()
return "tinyllama" in model_path.lower()

def get_default_conv_template(self, model_path: str) -> Conversation:
return get_conv_template("TinyLlama")
Expand Down

0 comments on commit 4d113fb

Please sign in to comment.