Fix indentation in load_variant (token + from_pretrained)

#1
by gberton - opened
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -294,8 +294,8 @@ def load_variant(name):
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
 
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