Ostris AI Toolkit Volume Init for Runpod

init.sh

#!/usr/bin/env bash
set -e

echo "Runpod init starting..."

# Create persistent dirs
mkdir -p /mnt/{datasets,models,output,configs,hf,cache}

link_dir () {
  TARGET=$1
  LINK=$2

  if [ -L "$LINK" ]; then
    rm "$LINK"
  elif [ -e "$LINK" ]; then
    rm -rf "$LINK"
  fi

  ln -s "$TARGET" "$LINK"
}

# Link ai-toolkit dirs
link_dir /mnt/datasets /app/ai-toolkit/datasets
link_dir /mnt/models   /app/ai-toolkit/models
link_dir /mnt/output   /app/ai-toolkit/output
link_dir /mnt/configs  /app/ai-toolkit/configs

# Cache persistence (huge speed win)
link_dir /mnt/hf    /root/.cache/huggingface
link_dir /mnt/cache /root/.cache

echo "Symlinks ready"
ls -lah /app/ai-toolkit

echo "Starting ai-toolkit..."
exec /start.sh