shuaikang commited on
Commit
14991de
1 Parent(s): 2bb84b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ def init_model():
29
  """
30
  global tokenizer, model, streamer, terminators
31
  tokenizer = AutoTokenizer.from_pretrained(
32
- model_path, local_files_only=True)
33
 
34
  model = AutoModelForCausalLM.from_pretrained(
35
  model_path,
 
29
  """
30
  global tokenizer, model, streamer, terminators
31
  tokenizer = AutoTokenizer.from_pretrained(
32
+ model_path, trust_remote_code=True)
33
 
34
  model = AutoModelForCausalLM.from_pretrained(
35
  model_path,