From 254ea0ca9a94dd87a92ffd49613b97233683af38 Mon Sep 17 00:00:00 2001 From: Alex Bason Date: Tue, 21 Jan 2025 11:37:48 -0500 Subject: [PATCH] Move torchvision to the torch extra --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 0e6645bc..b15cc8d4 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -20,12 +20,12 @@ dependencies = [ "pillow>=11.0.0", "markdownify>=0.14.1", "duckduckgo-search>=6.3.7", - "torchvision" ] [project.optional-dependencies] torch = [ "torch", + "torchvision", ] audio = [ "soundfile",