diff --git a/artifact_comfyui_download.sh b/artifact_comfyui_download.sh
index 5c16f9b..1010441 100755
--- a/artifact_comfyui_download.sh
+++ b/artifact_comfyui_download.sh
@@ -107,9 +107,16 @@ COMFYUI_DIR="${COMFYUI_DIR:-${HOME}/ComfyUI/models}"
 COMMAND="both"
 
 # HuggingFace token from environment or .env file
-if [[ -f "${PROJECT_ROOT}/.env" ]]; then
+# Try multiple locations for .env file
+if [[ -z "${HF_TOKEN}" ]] && [[ -f "${PROJECT_ROOT}/ai/.env" ]]; then
+    HF_TOKEN=$(grep ^HF_TOKEN "${PROJECT_ROOT}/ai/.env" | cut -d'=' -f2- | tr -d '"' | tr -d "'" || true)
+fi
+if [[ -z "${HF_TOKEN}" ]] && [[ -f "${PROJECT_ROOT}/.env" ]]; then
     HF_TOKEN=$(grep ^HF_TOKEN "${PROJECT_ROOT}/.env" | cut -d'=' -f2- | tr -d '"' | tr -d "'" || true)
 fi
+if [[ -z "${HF_TOKEN}" ]] && [[ -f "/workspace/ai/.env" ]]; then
+    HF_TOKEN=$(grep ^HF_TOKEN "/workspace/ai/.env" | cut -d'=' -f2- | tr -d '"' | tr -d "'" || true)
+fi
 HF_TOKEN="${HF_TOKEN:-}"
 
 # ============================================================================