services:
  qwen-lora-train:
    image: nvcr.io/nvidia/pytorch:25.12-py3
    container_name: qwen-lora-train
    restart: unless-stopped
    command: ["sleep", "infinity"]
    working_dir: /workspace/ai-projects/qwen3-6-lora
    shm_size: "16gb"
    volumes:
      - /mnt/docker-nas/projects/ai-projects:/workspace/ai-projects
      - /home/aleleba/ft-models:/workspace/ft-models
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]