diff --git a/installer.py b/installer.py index e03512301..c1ec1f177 100644 --- 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 index 5deab1282..7bc259391 100644 --- 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,