Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
752e7b9
1
Parent(s):
217b29d
Change the CACHE PATH of HF and transformers lib to /tmp/xx.
Browse files
app.py
CHANGED
|
@@ -19,10 +19,10 @@ import zipfile
|
|
| 19 |
|
| 20 |
|
| 21 |
transformers.utils.move_cache()
|
| 22 |
-
os.environ["HUGGINGFACE_HUB_CACHE"] = "/
|
| 23 |
-
os.environ["HF_HOME"] = "/
|
| 24 |
-
os.environ["TRANSFORMERS_CACHE"] = "/
|
| 25 |
-
os.makedirs("/
|
| 26 |
|
| 27 |
|
| 28 |
torch.set_float32_matmul_precision('high')
|
|
|
|
| 19 |
|
| 20 |
|
| 21 |
transformers.utils.move_cache()
|
| 22 |
+
os.environ["HUGGINGFACE_HUB_CACHE"] = "/tmp/hf_cache"
|
| 23 |
+
os.environ["HF_HOME"] = "/tmp/hf_cache"
|
| 24 |
+
os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
|
| 25 |
+
os.makedirs("/tmp/hf_cache", exist_ok=True)
|
| 26 |
|
| 27 |
|
| 28 |
torch.set_float32_matmul_precision('high')
|