Explorar o código

fix: update torch dependency version for compatibility with latest features

myhloli hai 3 meses
pai
achega
8c8ac2c667
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      pyproject.toml

+ 1 - 1
pyproject.toml

@@ -68,7 +68,7 @@ pipeline = [
     "shapely>=2.0.7,<3",
     "pyclipper>=1.3.0,<2",
     "omegaconf>=2.3.0,<3",
-    "torch>=2.2.2,!=2.5.0,!=2.5.1,<3",
+    "torch>=2.6.0,<3",
     "torchvision",
     "transformers>=4.49.0,!=4.51.0,<5.0.0",
 ]