Merge pull request #3617 from vladmandic/master

Pin ultralytics
vladmandic · Dec 5, 2024 · 6f6ba0b · 6f6ba0b
2 parents ca06002 + f79e85e
commit 6f6ba0b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/installer.py b/installer.py
@@ -1002,7 +1002,7 @@ def install_optional():
     install('optimum-quanto', ignore=True)
     install('bitsandbytes', ignore=True)
     install('pynvml', ignore=True)
-    install('ultralytics', ignore=True)
+    install('ultralytics==8.3.40', ignore=True)
     install('Cython', ignore=True)
     install('insightface', ignore=True) # problematic build
     install('nncf==2.7.0', ignore=True, no_deps=True) # requires older pandas

diff --git a/modules/postprocess/yolo.py b/modules/postprocess/yolo.py
@@ -63,7 +63,7 @@ def enumerate(self):
 
     def dependencies(self):
         import installer
-        installer.install('ultralytics', ignore=True, quiet=True)
+        installer.install('ultralytics==8.3.40', ignore=True, quiet=True)
 
     def predict(
             self,