0403c6f47e
Co-authored-by: SolitaryThinker <wlsaidhi@gmail.com> Co-authored-by: Matthew Noto <99706358+RandNMR73@users.noreply.github.com> Co-authored-by: XOR-op <17672363+XOR-op@users.noreply.github.com> Co-authored-by: Zhang Peiyuan <42993249+jzhang38@users.noreply.github.com>
76 lines
2.2 KiB
Bash
Executable File
76 lines
2.2 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
: "${CEREBRAS_API_KEY:?CEREBRAS_API_KEY not set on host}"
|
|
: "${GROQ_API_KEY:?GROQ_API_KEY not set on host}"
|
|
|
|
SCRIPT_DIR="$(cd -- "$(dirname -- "${BASH_SOURCE[0]}")" && pwd)"
|
|
DREAMVERSE_ROOT="$(cd -- "${SCRIPT_DIR}/.." && pwd)"
|
|
|
|
IMAGE="${DREAMVERSE_IMAGE:-dreamverse:dev}"
|
|
PORT="${BACKEND_PORT:-8009}"
|
|
HF_CACHE="${HF_HOME:-$HOME/.cache/huggingface}"
|
|
OUTPUTS_DIR="${DREAMVERSE_OUTPUTS_DIR:-${DREAMVERSE_ROOT}/outputs}"
|
|
NAME="${DREAMVERSE_NAME:-dreamverse}"
|
|
TIMEOUT_SECONDS="${DREAMVERSE_SMOKE_TIMEOUT_SECONDS:-1200}"
|
|
POLL_SECONDS="${DREAMVERSE_SMOKE_POLL_SECONDS:-5}"
|
|
GPU_REQUEST="${DREAMVERSE_DOCKER_GPUS:-all}"
|
|
|
|
mkdir -p "${HF_CACHE}" "${OUTPUTS_DIR}"
|
|
|
|
docker rm -f "${NAME}" >/dev/null 2>&1 || true
|
|
|
|
env_args=(
|
|
-e "CEREBRAS_API_KEY=${CEREBRAS_API_KEY}"
|
|
-e "GROQ_API_KEY=${GROQ_API_KEY}"
|
|
-e "ENABLE_TORCH_COMPILE=${ENABLE_TORCH_COMPILE:-0}"
|
|
)
|
|
[[ -n "${FASTVIDEO_GPU_COUNT:-}" ]] && env_args+=(-e "FASTVIDEO_GPU_COUNT=${FASTVIDEO_GPU_COUNT}")
|
|
|
|
container_id="$(
|
|
docker run -d --rm --gpus "${GPU_REQUEST}" --init \
|
|
-p "${PORT}:8009" \
|
|
"${env_args[@]}" \
|
|
-v "${HF_CACHE}:/root/.cache/huggingface" \
|
|
-v "${OUTPUTS_DIR}:/var/lib/dreamverse/outputs" \
|
|
--name "${NAME}" \
|
|
"${IMAGE}"
|
|
)"
|
|
|
|
cleanup() {
|
|
if [[ "${DREAMVERSE_KEEP_CONTAINER:-0}" != "1" ]]; then
|
|
docker rm -f "${NAME}" >/dev/null 2>&1 || true
|
|
fi
|
|
}
|
|
trap cleanup EXIT
|
|
|
|
wait_for_endpoint() {
|
|
local path="$1"
|
|
local label="$2"
|
|
local deadline=$((SECONDS + TIMEOUT_SECONDS))
|
|
local url="http://127.0.0.1:${PORT}${path}"
|
|
|
|
echo "Waiting for ${label} at ${url}"
|
|
while (( SECONDS < deadline )); do
|
|
if curl -fsS "${url}" >/dev/null 2>&1; then
|
|
echo "${label} ok"
|
|
return 0
|
|
fi
|
|
if ! docker ps --format '{{.Names}}' | grep -qx "${NAME}"; then
|
|
echo "Container exited before ${label} became healthy." >&2
|
|
docker logs "${container_id}" >&2 || true
|
|
return 1
|
|
fi
|
|
sleep "${POLL_SECONDS}"
|
|
done
|
|
|
|
echo "Timed out waiting for ${label}." >&2
|
|
docker logs "${container_id}" >&2 || true
|
|
return 1
|
|
}
|
|
|
|
wait_for_endpoint "/healthz" "healthz"
|
|
wait_for_endpoint "/readyz" "readyz"
|
|
|
|
echo "Dreamverse Docker smoke passed for ${IMAGE} on host port ${PORT}."
|