services:
  scenema-audio:
    build: .
    ports:
      - "8000:8000"
    environment:
      - HF_TOKEN=${HF_TOKEN}
      - GEMMA_QUANTIZE=nf4
      - ENABLE_GRADIO=1
      - AUDIO_CKPT=/app/models/scenema-audio-transformer-int8.safetensors
      - VAE_ENCODER_CKPT=/app/models/scenema-audio-vae-encoder.safetensors
      - PIPELINE_CKPT=/app/models/scenema-audio-pipeline.safetensors
      - GEMMA_ROOT=/app/models/gemma-3-12b-it
      - MELBAND_MODEL_PATH=/app/models/MelBandRoformer_fp16.safetensors
      - SEEDVC_PATH=/app/seed-vc
      - MODEL_DIR=/app/models
      - PORT=8000
    volumes:
      - model-cache:/app/models
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]

volumes:
  model-cache: