Spaces:
Sleeping
Sleeping
File size: 823 Bytes
f580891 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
# prepare_env.py
import os
TMP_PATH = "/tmp"
os.environ["HF_HOME"] = os.path.join(TMP_PATH, "huggingface")
os.environ["TRANSFORMERS_CACHE"] = os.environ["HF_HOME"] + "/transformers"
os.environ["HF_DATASETS_CACHE"] = os.environ["HF_HOME"] + "/datasets"
os.environ["HF_METRICS_CACHE"] = os.environ["HF_HOME"] + "/metrics"
os.environ["MPLCONFIGDIR"] = os.path.join(TMP_PATH, "matplotlib")
os.environ["TORCH_HOME"] = os.path.join(TMP_PATH, "torch")
os.environ["XDG_CACHE_HOME"] = os.path.join(TMP_PATH, "xdg-cache")
os.environ["HOME"] = TMP_PATH
# Création des dossiers si nécessaires
for path in [
os.environ["HF_HOME"],
os.environ["MPLCONFIGDIR"],
os.environ["TORCH_HOME"],
os.environ["XDG_CACHE_HOME"],
TMP_PATH
]:
os.makedirs(path, exist_ok=True)
print("✅ Environnement temporaire prêt.")
|