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):