services:
  smollora:
    build: .
    image: smollora
    container_name: smollora
    # runtime: nvidia  # uncomment if nvidia-container-toolkit is installed
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]
    volumes:
      - /srv/smollora/data:/data           # processed datasets
      - /srv/smollora/output:/output       # LoRA adapter output
      - /srv/smollora/hf-cache:/root/.cache/huggingface  # model cache
    environment:
      - MODEL=HuggingFaceTB/SmolLM3-3B
      - DATA_DIR=/data
      - OUTPUT_DIR=/output
      - EPOCHS=3
      - BATCH_SIZE=4
      - LR=2e-4
      - LORA_R=16
      - MAX_LENGTH=4096
      - HF_TOKEN=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
      - SKIP_PREP=0