xeze/embedding-api repository overview

services:
  embedding-api:
    image: xeze/embedding-api:latest
    environment:
      - EMBEDDING_MODEL_NAME=${EMBEDDING_MODEL_NAME}
    ports:
      - "8000:8000"
    volumes:
      - hf-cache:/root/.cache/huggingface

volumes:
  hf-cache:

services:
  embedding-api:
    image: xeze/embedding-api:gpu
    environment:
      - EMBEDDING_MODEL_NAME=${EMBEDDING_MODEL_NAME}
    ports:
      - "8000:8000"
    volumes:
      - hf-cache:/root/.cache/huggingface
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]

volumes:
  hf-cache:

# .env
EMBEDDING_MODEL_NAME=BAAI/bge-small-en-v1.5