Hjgugugjhuhjggg commited on
Commit
5d4a408
·
verified ·
1 Parent(s): e6982de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -97,10 +97,10 @@ class S3ModelLoader:
97
  try:
98
  config = AutoConfig.from_pretrained(model_name)
99
  tokenizer = AutoTokenizer.from_pretrained(
100
- model_name, config=config
101
  )
102
  model = AutoModelForSeq2SeqLM.from_pretrained(
103
- model_name, config=config
104
  )
105
 
106
  if tokenizer.eos_token_id is not None and \
 
97
  try:
98
  config = AutoConfig.from_pretrained(model_name)
99
  tokenizer = AutoTokenizer.from_pretrained(
100
+ model_name, config=config, token=HUGGINGFACE_HUB_TOKEN
101
  )
102
  model = AutoModelForSeq2SeqLM.from_pretrained(
103
+ model_name, config=config, token=HUGGINGFACE_HUB_TOKEN
104
  )
105
 
106
  if tokenizer.eos_token_id is not None and \