Skip to content
Prev Previous commit
Next Next commit
Update fastchat/model/model_adapter.py
Co-authored-by: Lianmin Zheng <lianminzheng@gmail.com>
  • Loading branch information
wangzhen263 and merrymercy authored Sep 18, 2023
commit 3cf629bd0668afb440c542932be1418e7745b889
2 changes: 1 addition & 1 deletion fastchat/model/model_adapter.py
Original file line number Diff line number Diff line change
Expand Up @@ -620,7 +620,7 @@ class GoogleFlanAdapter(BaseModelAdapter):

def match(self, model_path: str):
return any(
model_path in model_str for model_str in ["flan-", "fastchat-t5", "codet5p"]
model_str in model_path.lower() for model_str in ["flan-", "fastchat-t5", "codet5p"]
)

def load_model(self, model_path: str, from_pretrained_kwargs: dict):
Expand Down