Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
cbc80cf
1
Parent(s):
4396298
Specify HF_MODULES_CACHE.
Browse files
app.py
CHANGED
@@ -23,7 +23,8 @@ hf_cache_path = '/tmp/hf_cache'
|
|
23 |
os.environ["HUGGINGFACE_HUB_CACHE"] = hf_cache_path
|
24 |
os.environ["HF_HOME"] = hf_cache_path
|
25 |
os.environ["TRANSFORMERS_CACHE"] = hf_cache_path
|
26 |
-
os.
|
|
|
27 |
|
28 |
|
29 |
torch.set_float32_matmul_precision('high')
|
|
|
23 |
os.environ["HUGGINGFACE_HUB_CACHE"] = hf_cache_path
|
24 |
os.environ["HF_HOME"] = hf_cache_path
|
25 |
os.environ["TRANSFORMERS_CACHE"] = hf_cache_path
|
26 |
+
os.environ["HF_MODULES_CACHE"] = os.path.join("/tmp/hf_cache", "modules")
|
27 |
+
os.makedirs(os.path.join("/tmp/hf_cache", "modules"), exist_ok=True)
|
28 |
|
29 |
|
30 |
torch.set_float32_matmul_precision('high')
|