Spaces:
Running
on
Zero
Running
on
Zero
Upload app.py
Browse files
app.py
CHANGED
|
@@ -31,6 +31,8 @@ import shutil
|
|
| 31 |
import yaml
|
| 32 |
import numpy as np
|
| 33 |
from huggingface_hub import hf_hub_download
|
|
|
|
|
|
|
| 34 |
|
| 35 |
dtype = torch.bfloat16
|
| 36 |
device = "cuda"
|
|
|
|
| 31 |
import yaml
|
| 32 |
import numpy as np
|
| 33 |
from huggingface_hub import hf_hub_download
|
| 34 |
+
import huggingface_hub
|
| 35 |
+
huggingface_hub.login(os.getenv('HF_TOKEN'))
|
| 36 |
|
| 37 |
dtype = torch.bfloat16
|
| 38 |
device = "cuda"
|