From 1c853b725e6cb9451f1e699ab3aee7b2b5751afd Mon Sep 17 00:00:00 2001 From: Eugene Rakhmatulin Date: Wed, 25 Feb 2026 16:38:04 -0800 Subject: [PATCH] allows to use $HF_HOME as huggingface cache directory, closes #68 --- launch-cluster.sh | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/launch-cluster.sh b/launch-cluster.sh index c014d6b..b677f2c 100755 --- a/launch-cluster.sh +++ b/launch-cluster.sh @@ -3,8 +3,9 @@ # Default Configuration IMAGE_NAME="vllm-node" DEFAULT_CONTAINER_NAME="vllm_node" +HF_CACHE_DIR="${HF_HOME:-$HOME/.cache/huggingface}" # Modify these if you want to pass additional docker args or set VLLM_SPARK_EXTRA_DOCKER_ARGS variable -DOCKER_ARGS="-e NCCL_IGNORE_CPU_AFFINITY=1 -v $HOME/.cache/huggingface:/root/.cache/huggingface" +DOCKER_ARGS="-e NCCL_IGNORE_CPU_AFFINITY=1 -v $HF_CACHE_DIR:/root/.cache/huggingface" # Append additional arguments from environment variable if [[ -n "$VLLM_SPARK_EXTRA_DOCKER_ARGS" ]]; then