Fix PathLike model_id load.

HIT-SCIR · Feb 18, 2025 · e1b4e34 · e1b4e34
1 parent 797c420
commit e1b4e34
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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):