fix: updated .cache dirs
Browse files- gradio_app.py +6 -0
- 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 |
|