Orion-zhen commited on
Commit
9579fc0
·
verified ·
1 Parent(s): dd1b211

trust remote code

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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)