From 72b381fdf8b67465d91ba1cb4768c7be1a34ffb4 Mon Sep 17 00:00:00 2001 From: Andrew Lapp Date: Thu, 10 Oct 2024 14:22:54 -0400 Subject: [PATCH] make torch a default requirement --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index f603b918a..fa7005afd 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -40,6 +40,7 @@ dependencies = [ "typing_extensions", "pycountry", "airportsdata", + "torch", "outlines_core==0.1.14", ] dynamic = ["version"] @@ -61,7 +62,6 @@ test = [ "huggingface_hub", "openai>=1.0.0", "vllm; sys_platform != 'darwin'", - "torch", "transformers", "pillow", "exllamav2",