bingyic commited on
Commit
b99ab8d
·
verified ·
1 Parent(s): f8fe515

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -294,7 +294,8 @@ def load_variant(name):
294
  global _model
295
  if _model["name"] == name:
296
  return
297
- dpt = AutoModel.from_pretrained(VARIANTS[name], trust_remote_code=True, token=os.environ.get("HF_TOKEN"))
 
298
  dpt.eval()
299
  dpt._get_backbone() # trigger backbone download
300
  backbone = dpt._backbone
 
294
  global _model
295
  if _model["name"] == name:
296
  return
297
+ token = os.environ.get("HF_TIPSv2") or os.environ.get("HF_TOKEN")
298
+ dpt = AutoModel.from_pretrained(VARIANTS[name], trust_remote_code=True, token=token)
299
  dpt.eval()
300
  dpt._get_backbone() # trigger backbone download
301
  backbone = dpt._backbone