From 33ceaa0eb8ef033a341375e20aca9f42fee56008 Mon Sep 17 00:00:00 2001 From: Songyang Zhang Date: Tue, 30 Jul 2024 09:37:50 +0800 Subject: [PATCH] [Bug] Fix bug in turbomind (#1377) --- opencompass/models/turbomind.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/opencompass/models/turbomind.py b/opencompass/models/turbomind.py index 33e1b984..af769c28 100644 --- a/opencompass/models/turbomind.py +++ b/opencompass/models/turbomind.py @@ -61,7 +61,8 @@ class TurboMindModel(BaseModel): from lmdeploy.messages import TurbomindEngineConfig engine_config = TurbomindEngineConfig(**engine_config) self.logger = get_logger() - assert os.path.exists(path), '{} is not existist'.format(path) + if path.startswith('/') or path.startswith('.'): + assert os.path.exists(path), '{} is not existist'.format(path) tm_model = TurboMind.from_pretrained(path, engine_config=engine_config) self.tokenizer = tm_model.tokenizer self.generators = [