Prechádzať zdrojové kódy

fix: update torch dependency version for compatibility with latest features

myhloli 3 mesiacov pred
rodič
commit
8c8ac2c667
1 zmenil súbory, kde vykonal 1 pridanie a 1 odobranie
  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",
 ]