Update wanForProduction.sh
Browse files- wanForProduction.sh +4 -7
wanForProduction.sh
CHANGED
|
@@ -24,6 +24,9 @@ COMFY_PORT="${COMFY_PORT:-8818}"
|
|
| 24 |
CIVITAI_TOKEN="${CIVITAI_TOKEN:-}"
|
| 25 |
HF_TOKEN="${HF_TOKEN:-}"
|
| 26 |
|
|
|
|
|
|
|
|
|
|
| 27 |
# Performance
|
| 28 |
export MAX_JOBS="${MAX_JOBS:-32}"
|
| 29 |
export HF_HUB_ENABLE_HF_TRANSFER=1
|
|
@@ -126,13 +129,7 @@ download_hf() {
|
|
| 126 |
success=false
|
| 127 |
|
| 128 |
if command -v hf >/dev/null 2>&1; then
|
| 129 |
-
if HF_HUB_ENABLE_HF_TRANSFER=1 hf download "$repo_id" "$rest" --local-dir "$temp_dir" --
|
| 130 |
-
success=true
|
| 131 |
-
fi
|
| 132 |
-
fi
|
| 133 |
-
|
| 134 |
-
if [[ "$success" == false ]] && command -v huggingface-cli >/dev/null 2>&1; then
|
| 135 |
-
if HF_HUB_ENABLE_HF_TRANSFER=1 huggingface-cli download "$repo_id" "$rest" --local-dir "$temp_dir" --local-dir-use-symlinks False; then
|
| 136 |
success=true
|
| 137 |
fi
|
| 138 |
fi
|
|
|
|
| 24 |
CIVITAI_TOKEN="${CIVITAI_TOKEN:-}"
|
| 25 |
HF_TOKEN="${HF_TOKEN:-}"
|
| 26 |
|
| 27 |
+
# Suprimir avisos do pip quando executado como root
|
| 28 |
+
export PIP_ROOT_USER_ACTION=ignore
|
| 29 |
+
|
| 30 |
# Performance
|
| 31 |
export MAX_JOBS="${MAX_JOBS:-32}"
|
| 32 |
export HF_HUB_ENABLE_HF_TRANSFER=1
|
|
|
|
| 129 |
success=false
|
| 130 |
|
| 131 |
if command -v hf >/dev/null 2>&1; then
|
| 132 |
+
if HF_HUB_ENABLE_HF_TRANSFER=1 hf download "$repo_id" "$rest" --local-dir "$temp_dir" --quiet; then
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 133 |
success=true
|
| 134 |
fi
|
| 135 |
fi
|