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"
|
||||
|
||||
# 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:-}"
|
||||
|
||||
# ============================================================================
|
||||
|
||||
Reference in New Issue
Block a user