p1atdev commited on
Commit
d0c284e
1 Parent(s): 209dbde

fix: set use_auth_token

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,7 +25,7 @@ tokenizer = AutoTokenizer.from_pretrained(
25
  model = {
26
  "default": AutoModelForCausalLM.from_pretrained(
27
  MODEL_NAME,
28
- use_auth_token=HF_READ_TOKEN,
29
  ),
30
  "ort": ORTModelForCausalLM.from_pretrained(
31
  MODEL_NAME,
@@ -34,7 +34,7 @@ model = {
34
  "ort_qantized": ORTModelForCausalLM.from_pretrained(
35
  MODEL_NAME,
36
  file_name="model_quantized.onnx",
37
- token=HF_READ_TOKEN,
38
  ),
39
  }
40
 
 
25
  model = {
26
  "default": AutoModelForCausalLM.from_pretrained(
27
  MODEL_NAME,
28
+ token=HF_READ_TOKEN,
29
  ),
30
  "ort": ORTModelForCausalLM.from_pretrained(
31
  MODEL_NAME,
 
34
  "ort_qantized": ORTModelForCausalLM.from_pretrained(
35
  MODEL_NAME,
36
  file_name="model_quantized.onnx",
37
+ use_auth_token=HF_READ_TOKEN,
38
  ),
39
  }
40