make torch a default requirement

dottxt-ai · Oct 10, 2024 · 72b381f · 72b381f
1 parent 5ccf4e5
commit 72b381f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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",