From bc862f1d2502a97ed8cbba410187cd5a9fbde649 Mon Sep 17 00:00:00 2001 From: Ashwin Bharambe Date: Mon, 3 Feb 2025 13:52:39 -0800 Subject: [PATCH] Make sure torch + torchvision go together as deps --- llama_stack/providers/registry/vector_io.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llama_stack/providers/registry/vector_io.py b/llama_stack/providers/registry/vector_io.py index df7b7f4b3..2d7c02d86 100644 --- a/llama_stack/providers/registry/vector_io.py +++ b/llama_stack/providers/registry/vector_io.py @@ -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", ]