Spaces:
Running
Running
trust remote code
Browse files
app.py
CHANGED
@@ -31,7 +31,7 @@ def tokenize_text(model_name, text):
|
|
31 |
try:
|
32 |
# 加载tokenizer
|
33 |
model_path = os.path.join("models", model_name)
|
34 |
-
tokenizer = AutoTokenizer.from_pretrained(model_path, device_map="cpu")
|
35 |
|
36 |
# Tokenize处理
|
37 |
input_ids = tokenizer.encode(text, add_special_tokens=True)
|
|
|
31 |
try:
|
32 |
# 加载tokenizer
|
33 |
model_path = os.path.join("models", model_name)
|
34 |
+
tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True, device_map="cpu")
|
35 |
|
36 |
# Tokenize处理
|
37 |
input_ids = tokenizer.encode(text, add_special_tokens=True)
|