Explorar el Código

fix: update torch dependency version for compatibility with latest features

myhloli hace 3 meses
padre
commit
8c8ac2c667
Se han modificado 1 ficheros con 1 adiciones y 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",
 ]