fix: check multiple locations for .env file
- Try PROJECT_ROOT/ai/.env first - Fallback to PROJECT_ROOT/.env - Fallback to /workspace/ai/.env - Fixes HF_TOKEN loading on RunPod 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
@@ -107,9 +107,16 @@ COMFYUI_DIR="${COMFYUI_DIR:-${HOME}/ComfyUI/models}"
|
|||||||
COMMAND="both"
|
COMMAND="both"
|
||||||
|
|
||||||
# HuggingFace token from environment or .env file
|
# 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)
|
HF_TOKEN=$(grep ^HF_TOKEN "${PROJECT_ROOT}/.env" | cut -d'=' -f2- | tr -d '"' | tr -d "'" || true)
|
||||||
fi
|
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:-}"
|
HF_TOKEN="${HF_TOKEN:-}"
|
||||||
|
|
||||||
# ============================================================================
|
# ============================================================================
|
||||||
|
|||||||
Reference in New Issue
Block a user