Remove hardcoded root path for HF models

onyx-dot-app · Dec 17, 2024 · b5d0df9 · b5d0df9
1 parent 48be633
commit b5d0df9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/backend/model_server/main.py b/backend/model_server/main.py
@@ -27,8 +27,8 @@
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 os.environ["HF_HUB_DISABLE_TELEMETRY"] = "1"
 
-HF_CACHE_PATH = Path("/root/.cache/huggingface/")
-TEMP_HF_CACHE_PATH = Path("/root/.cache/temp_huggingface/")
+HF_CACHE_PATH = Path(os.path.expanduser("~")) / ".cache/huggingface"
+TEMP_HF_CACHE_PATH = Path(os.path.expanduser("~")) / ".cache/temp_huggingface"
 
 transformer_logging.set_verbosity_error()