From 557c05716c63766921078ac92c6a82328c693345 Mon Sep 17 00:00:00 2001 From: violetc <58360096+s-yh-china@users.noreply.github.com> Date: Tue, 18 Feb 2025 16:43:55 +0800 Subject: [PATCH] Fix PathLike model_id load. (#722) --- python/interface/ltp/interface.py | 2 +- python/interface/ltp/mixin.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/python/interface/ltp/interface.py b/python/interface/ltp/interface.py index 6faf9125..95b2a210 100644 --- a/python/interface/ltp/interface.py +++ b/python/interface/ltp/interface.py @@ -85,7 +85,7 @@ def LTP( model_id = pretrained_model_name_or_path revision = None - if len(model_id.split("@")) == 2: + if isinstance(model_id, str) and len(model_id.split("@")) == 2: model_id, revision = model_id.split("@") if os.path.isdir(model_id) and CONFIG_NAME in os.listdir(model_id): diff --git a/python/interface/ltp/mixin.py b/python/interface/ltp/mixin.py index 097aaeaa..2910e6fa 100644 --- a/python/interface/ltp/mixin.py +++ b/python/interface/ltp/mixin.py @@ -107,7 +107,7 @@ def from_pretrained( model_id = pretrained_model_name_or_path revision = None - if len(model_id.split("@")) == 2: + if isinstance(model_id, str) and len(model_id.split("@")) == 2: model_id, revision = model_id.split("@") if os.path.isdir(model_id) and CONFIG_NAME in os.listdir(model_id):