mirror of
https://github.com/coleam00/Archon.git
synced 2025-12-30 21:49:30 -05:00
Reduced the size of sentence-transformers by making it CPU only, including reranking by default now (#534)
This commit is contained in:
@@ -15,7 +15,7 @@ COPY pyproject.toml .
|
||||
# Install server dependencies to a virtual environment using uv
|
||||
RUN uv venv /venv && \
|
||||
. /venv/bin/activate && \
|
||||
uv pip install --group server
|
||||
uv pip install --group server --group server-reranking
|
||||
|
||||
# Runtime stage
|
||||
FROM python:3.12-slim
|
||||
|
||||
@@ -7,6 +7,16 @@ requires-python = ">=3.12"
|
||||
# Base dependencies - empty since we're using dependency groups
|
||||
dependencies = []
|
||||
|
||||
# PyTorch CPU-only index configuration
|
||||
[[tool.uv.index]]
|
||||
name = "pytorch-cpu"
|
||||
url = "https://download.pytorch.org/whl/cpu"
|
||||
explicit = true
|
||||
|
||||
# Sources configuration to use CPU-only PyTorch
|
||||
[tool.uv.sources]
|
||||
torch = [{ index = "pytorch-cpu" }]
|
||||
|
||||
[dependency-groups]
|
||||
# Development dependencies for linting and testing
|
||||
dev = [
|
||||
|
||||
Reference in New Issue
Block a user