mirror of
https://github.com/aaronpo97/the-biergarten-app.git
synced 2026-06-01 01:54:00 +00:00
60 lines
1.9 KiB
Bash
60 lines
1.9 KiB
Bash
#!/bin/bash
|
|
set -e
|
|
|
|
# Configuration / Defaults
|
|
MODEL_PATH="${BIERGARTEN_MODEL_PATH:-/workspace/models/google_gemma-4-E4B-it-Q6_K.gguf}"
|
|
OUTPUT_DIR="${BIERGARTEN_OUTPUT_DIR:-/workspace/output}"
|
|
LOG_PATH="${BIERGARTEN_LOG_PATH:-/workspace/logs/pipeline.log}"
|
|
EXECUTABLE="/app/build/biergarten-pipeline"
|
|
PROMPT_DIR="/app/prompts"
|
|
|
|
echo "--- Starting Biergarten Pipeline Environment Check ---"
|
|
|
|
# 1. Ensure volume mount directories exist
|
|
mkdir -p "$OUTPUT_DIR"
|
|
mkdir -p "$(dirname "$LOG_PATH")"
|
|
|
|
# 3. Build the command arguments
|
|
ARGS=(
|
|
"--model" "$MODEL_PATH"
|
|
"--prompt-dir" "$PROMPT_DIR"
|
|
"--output" "$OUTPUT_DIR"
|
|
"--log-path" "$LOG_PATH"
|
|
)
|
|
|
|
# Optional hyperparameters
|
|
[[ -n "$BIERGARTEN_TEMPERATURE" ]] && ARGS+=("--temperature" "$BIERGARTEN_TEMPERATURE")
|
|
[[ -n "$BIERGARTEN_TOP_P" ]] && ARGS+=("--top-p" "$BIERGARTEN_TOP_P")
|
|
[[ -n "$BIERGARTEN_TOP_K" ]] && ARGS+=("--top-k" "$BIERGARTEN_TOP_K")
|
|
[[ -n "$BIERGARTEN_N_CTX" ]] && ARGS+=("--n-ctx" "$BIERGARTEN_N_CTX")
|
|
[[ -n "$BIERGARTEN_SEED" ]] && ARGS+=("--seed" "$BIERGARTEN_SEED")
|
|
[[ -n "$BIERGARTEN_GL_LAYERS" ]] && ARGS+=("--n-gpu-layers" "$BIERGARTEN_GL_LAYERS")
|
|
|
|
# Append any extra custom args
|
|
if [[ -n "$BIERGARTEN_EXTRA_ARGS" ]]; then
|
|
ARGS+=($BIERGARTEN_EXTRA_ARGS)
|
|
fi
|
|
|
|
echo "--- Executing: $EXECUTABLE ${ARGS[*]} ---"
|
|
# 2. Ensure model directory exists
|
|
mkdir -p "$(dirname "$MODEL_PATH")"
|
|
|
|
# 3. Download model if missing
|
|
if [ ! -f "$MODEL_PATH" ]; then
|
|
echo "Model not found. Downloading..."
|
|
|
|
curl -L \
|
|
-o "$MODEL_PATH" \
|
|
"https://huggingface.co/bartowski/google_gemma-4-E4B-it-GGUF/resolve/main/google_gemma-4-E4B-it-Q6_K.gguf?download=true"
|
|
|
|
echo "Download complete."
|
|
fi
|
|
|
|
# 4. Verify model exists
|
|
if [ ! -f "$MODEL_PATH" ]; then
|
|
echo "ERROR: Model still not found after download attempt."
|
|
exit 1
|
|
fi
|
|
# Execute the binary directly, replacing the shell process
|
|
exec "$EXECUTABLE" "${ARGS[@]}"
|