diff --git a/pyproject.toml b/pyproject.toml index 48eb38f..0a1b5d0 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -13,8 +13,9 @@ classifiers = [ ] requires-python = ">=3.10" dependencies = [ - "torch >= 2.6", + "torch <= 2.8", "torchvision", + "transformers", "numpy", "safetensors", "gguf",