name: unsloth

services:
  unsloth:
    image: unsloth/unsloth:latest
    container_name: unsloth
    restart: unless-stopped
    runtime: nvidia
    environment:
      - JUPYTER_PASSWORD=${JUPYTER_PASSWORD}
      - USER_PASSWORD=${USER_PASSWORD}
      - JUPYTER_PORT=8888
      - NVIDIA_VISIBLE_DEVICES=all
    ports:
      - "8888:8888"
    volumes:
      - ./work:/workspace/work
      - unsloth-cache:/home/unsloth/.cache
    networks:
      - traefik_proxy
      - default
    labels:
      - "traefik.enable=true"
      - "traefik.docker.network=traefik_proxy"
      - "traefik.http.routers.unsloth.rule=Host(`unsloth.kzbrd.ru`)"
      - "traefik.http.routers.unsloth.entrypoints=websecure"
      - "traefik.http.routers.unsloth.tls=true"
      - "traefik.http.routers.unsloth.tls.certresolver=myresolver"
      - "traefik.http.services.unsloth.loadbalancer.server.port=8000"
      - "com.centurylinklabs.watchtower.enable=true"

networks:
  traefik_proxy:
    external: true
  default:
    driver: bridge

volumes:
  unsloth-cache: