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:
@@ -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