ZhengPeng7 commited on
Commit
cbc80cf
·
1 Parent(s): 4396298

Specify HF_MODULES_CACHE.

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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.makedirs(hf_cache_path, exist_ok=True)
 
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')