Spaces:
Running
on
Zero
Running
on
Zero
AlekseyCalvin
commited on
Commit
•
48e7080
1
Parent(s):
2a30da1
Update app.py
Browse files
app.py
CHANGED
@@ -23,10 +23,10 @@ import warnings
|
|
23 |
import safetensors.torch
|
24 |
|
25 |
|
26 |
-
cache_path = path.join(path.dirname(path.abspath(__file__)), "models")
|
27 |
-
os.environ["TRANSFORMERS_CACHE"] = cache_path
|
28 |
-
os.environ["HF_HUB_CACHE"] = cache_path
|
29 |
-
os.environ["HF_HOME"] = cache_path
|
30 |
|
31 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
32 |
|
|
|
23 |
import safetensors.torch
|
24 |
|
25 |
|
26 |
+
#cache_path = path.join(path.dirname(path.abspath(__file__)), "models")
|
27 |
+
#os.environ["TRANSFORMERS_CACHE"] = cache_path
|
28 |
+
#os.environ["HF_HUB_CACHE"] = cache_path
|
29 |
+
#os.environ["HF_HOME"] = cache_path
|
30 |
|
31 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
32 |
|