Spaces:
Running on Zero
Running on Zero
Fix indentation in load_variant (token + from_pretrained)
Browse files
app.py
CHANGED
|
@@ -294,8 +294,8 @@ def load_variant(name):
|
|
| 294 |
global _model
|
| 295 |
if _model["name"] == name:
|
| 296 |
return
|
| 297 |
-
|
| 298 |
-
|
| 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
|