feat: Configure cache environment variables for Hugging Face and update Docker settings

This commit is contained in:
JB
2025-10-06 15:50:32 -07:00
parent dc7a115e2e
commit c19050261c
4 changed files with 53 additions and 21 deletions
+5
View File
@@ -23,6 +23,11 @@ services:
ABOGEN_OUTPUT_DIR: "/data/outputs"
ABOGEN_OUTPUT_ROOT: "/data/outputs"
ABOGEN_TEMP_DIR: "/data/cache"
HOME: "/data"
XDG_CACHE_HOME: "/data/cache"
HF_HOME: "/data/cache/huggingface"
HUGGINGFACE_HUB_CACHE: "/data/cache/huggingface"
TRANSFORMERS_CACHE: "/data/cache/huggingface"
# --- GPU support -----------------------------------------------------
# These settings assume the NVIDIA Container Toolkit is installed.
# Leave them in place for GPU acceleration; comment out the entire block