Spaces:
Paused
Paused
Update builder.sh
Browse files- builder.sh +6 -0
builder.sh
CHANGED
|
@@ -1,6 +1,9 @@
|
|
| 1 |
#!/usr/bin/env bash
|
| 2 |
set -euo pipefail
|
| 3 |
|
|
|
|
|
|
|
|
|
|
| 4 |
echo "======================================================="
|
| 5 |
echo " ADUC-SDR — Builder (Triton + Apex + Q8 + FlashAttention + LayerNorm)"
|
| 6 |
echo "======================================================="
|
|
@@ -34,6 +37,9 @@ export CUDA_HOME="${CUDA_HOME:-/usr/local/cuda}"
|
|
| 34 |
|
| 35 |
# ---------------------- Dependências de build ----------------------
|
| 36 |
|
|
|
|
|
|
|
|
|
|
| 37 |
# Tags do ambiente
|
| 38 |
PY_TAG="$(python -c 'import sys; print(f"cp{sys.version_info[0]}{sys.version_info[1]}")' 2>/dev/null || echo cp310)"
|
| 39 |
TORCH_VER="$(python - <<'PY'
|
|
|
|
| 1 |
#!/usr/bin/env bash
|
| 2 |
set -euo pipefail
|
| 3 |
|
| 4 |
+
|
| 5 |
+
pip uninstall -y triton
|
| 6 |
+
|
| 7 |
echo "======================================================="
|
| 8 |
echo " ADUC-SDR — Builder (Triton + Apex + Q8 + FlashAttention + LayerNorm)"
|
| 9 |
echo "======================================================="
|
|
|
|
| 37 |
|
| 38 |
# ---------------------- Dependências de build ----------------------
|
| 39 |
|
| 40 |
+
|
| 41 |
+
|
| 42 |
+
|
| 43 |
# Tags do ambiente
|
| 44 |
PY_TAG="$(python -c 'import sys; print(f"cp{sys.version_info[0]}{sys.version_info[1]}")' 2>/dev/null || echo cp310)"
|
| 45 |
TORCH_VER="$(python - <<'PY'
|