phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

Misc fixes (#944)

Browse source 
- Make sure torch + torchvision go together as deps, otherwise bad stuff
happens
- Add a pre-commit for requirements.txt
This commit is contained in:
Ashwin Bharambe 2025-02-03 14:08:47 -08:00 • committed by GitHub
parent 0f14378135
commit f98efe68c9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 73 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/registry/vector_io.py
View file

@ -30,7 +30,7 @@ EMBEDDING_DEPS = [
# we need a better way to do this to identify potential conflicts, etc.
# for now, this lets us significantly reduce the size of the container which
# does not have any "local" inference code (and hence does not need GPU-enabled torch)
"torch --index-url https://download.pytorch.org/whl/cpu",
"torch torchvision --index-url https://download.pytorch.org/whl/cpu",
"sentence-transformers --no-deps",
]