udayl commited on
Commit
fc0739a
·
1 Parent(s): 4a3d440

fix: updated .cache dirs

Browse files
Files changed (2) hide show
  1. gradio_app.py +6 -0
  2. notebook_lm_kokoro.py +6 -0
gradio_app.py CHANGED
@@ -22,6 +22,12 @@ from notebook_lm_kokoro import (
22
  KPipeline,
23
  )
24
 
 
 
 
 
 
 
25
  warnings.filterwarnings("ignore")
26
  NUM_WORKERS = multiprocessing.cpu_count()
27
 
 
22
  KPipeline,
23
  )
24
 
25
+ os.environ["HF_HOME"] = "/tmp/huggingface"
26
+ os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface/transformers"
27
+ os.environ["XDG_CACHE_HOME"] = "/tmp/huggingface"
28
+ os.environ["TORCH_HOME"] = "/tmp/torch"
29
+ os.environ["MOSHI_CACHE_DIR"] = "/tmp/moshi"
30
+
31
  warnings.filterwarnings("ignore")
32
  NUM_WORKERS = multiprocessing.cpu_count()
33
 
notebook_lm_kokoro.py CHANGED
@@ -38,6 +38,12 @@ openai.api_key = os.getenv("OPENAI_API_KEY")
38
  # For OpenRouter compatibility, set the API base if provided.
39
  openai.api_base = os.getenv("OPENROUTER_API_BASE", "https://api.openai.com/v1")
40
 
 
 
 
 
 
 
41
  pdf = "1706.03762v7.pdf"
42
 
43
 
 
38
  # For OpenRouter compatibility, set the API base if provided.
39
  openai.api_base = os.getenv("OPENROUTER_API_BASE", "https://api.openai.com/v1")
40
 
41
+ os.environ["HF_HOME"] = "/tmp/huggingface"
42
+ os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface/transformers"
43
+ os.environ["XDG_CACHE_HOME"] = "/tmp/huggingface"
44
+ os.environ["TORCH_HOME"] = "/tmp/torch"
45
+ os.environ["MOSHI_CACHE_DIR"] = "/tmp/moshi"
46
+
47
  pdf = "1706.03762v7.pdf"
48
 
49