From 73621c9cd1bc092a575de024925450a436e02572 Mon Sep 17 00:00:00 2001 From: Alex Bason Date: Fri, 24 Jan 2025 08:31:10 -0500 Subject: [PATCH] Move torchvision to the torch extra (#297) --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 873d0b3..76762bf 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",