Skip to content

Commit

Permalink
Update model_adapter.py (#2895)
Browse files Browse the repository at this point in the history
  • Loading branch information
thavens authored Jan 8, 2024
1 parent 2dbb9f1 commit 06acba1
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 @@ -1975,7 +1975,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 06acba1

Please sign in to comment.