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"
|