tranquilkd commited on
Commit
fd4433f
·
1 Parent(s): e8395e2

update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -17,7 +17,8 @@ def load_model(model_path):
17
  model = AutoModelForCausalLM.from_pretrained(
18
  config.base_model_name_or_path,
19
  quantization_config=bnb_config,
20
- trust_remote_code=True
 
21
  )
22
  model.config.use_cache = False
23
  tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
 
17
  model = AutoModelForCausalLM.from_pretrained(
18
  config.base_model_name_or_path,
19
  quantization_config=bnb_config,
20
+ trust_remote_code=True,
21
+ token=os.environ["HF_TOKEN"],
22
  )
23
  model.config.use_cache = False
24
  tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)