CRYPTONEWS34 commited on
Commit
a3bfc0a
·
1 Parent(s): e00b7f3

Fix: set TRANSFORMERS_CACHE before model load

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -2,6 +2,7 @@ import os
2
 
3
  os.environ["HF_HOME"] = "/tmp/huggingface"
4
  os.environ["MPLCONFIGDIR"] = "/tmp/mplconfig"
 
5
 
6
  os.makedirs("/tmp/huggingface", exist_ok=True)
7
  os.makedirs("/tmp/mplconfig", exist_ok=True)
 
2
 
3
  os.environ["HF_HOME"] = "/tmp/huggingface"
4
  os.environ["MPLCONFIGDIR"] = "/tmp/mplconfig"
5
+ os.environ["TRANSFORMERS_CACHE"] = "./hf_cache"
6
 
7
  os.makedirs("/tmp/huggingface", exist_ok=True)
8
  os.makedirs("/tmp/mplconfig", exist_ok=True)