version: "3"
services:
  sillytavern-extras:
    runtime: nvidia
    image: cohee1207/sillytavern-extras
    build:
      context: ../
      dockerfile: docker/Dockerfile
      args:
        REQUIREMENTS: requirements.txt
        MODULES: caption,summarize,classify
#        REQUIREMENTS: requirements-complete.txt
#        MODULES: caption,summarize,classify,keywords,prompt,sd,tts,chromadb
    ports:
      - "5100:5100"
    environment:
      - NVIDIA_VISIBLE_DEVICES=all
    command: python server.py --enable-modules=caption,summarize,classify
#    command: python server.py --enable-modules=caption,summarize,classify,keywords,prompt,sd,tts,chromadb