Upload d.py
Browse files
src/d.py
CHANGED
@@ -1,7 +1,10 @@
|
|
1 |
import os
|
|
|
2 |
os.environ["HF_HOME"] = "/tmp/huggingface"
|
3 |
os.environ["HF_HUB_CACHE"] = "/tmp/huggingface/hub"
|
4 |
os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface/transformers"
|
|
|
|
|
5 |
|
6 |
import yt_dlp
|
7 |
from pydub import AudioSegment
|
|
|
1 |
import os
|
2 |
+
# Set Hugging Face and Torch cache directories to /tmp (must be before any other imports)
|
3 |
os.environ["HF_HOME"] = "/tmp/huggingface"
|
4 |
os.environ["HF_HUB_CACHE"] = "/tmp/huggingface/hub"
|
5 |
os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface/transformers"
|
6 |
+
os.environ["TORCH_HOME"] = "/tmp/torch"
|
7 |
+
os.environ["XDG_CACHE_HOME"] = "/tmp/xdg_cache"
|
8 |
|
9 |
import yt_dlp
|
10 |
from pydub import AudioSegment
|