#!/bin/bash
set -e

echo "=== Wan2.2 Smoke Training ==="
echo ""

# Install dependencies
echo "Installing dependencies..."
pip install -r requirements.txt

# Run training
echo ""
python3 train.py \
  --data-dir . \
  --output-dir ./output \
  --lora-rank 16 \
  --epochs 1 \
  --learning-rate 2e-5

echo ""
echo "=== Training Complete ==="
echo "Checkpoints saved to: ./output"