HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Remove onnxslim==0.1.34 pin (#16974)

Browse source
This commit is contained in:
Glenn Jocher 2024-10-17 02:48:28 +02:00 • committed by GitHub
parent 3484daa1b5
commit 3204a2a6a7
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/engine/exporter.py
View file

@ -398,7 +398,7 @@ class Exporter:
"""YOLO ONNX export.""" """YOLO ONNX export."""
requirements = ["onnx>=1.12.0"] requirements = ["onnx>=1.12.0"]
if self.args.simplify: if self.args.simplify:
requirements += ["onnxslim==0.1.34", "onnxruntime" + ("-gpu" if torch.cuda.is_available() else "")] requirements += ["onnxslim", "onnxruntime" + ("-gpu" if torch.cuda.is_available() else "")]
check_requirements(requirements) check_requirements(requirements)
import onnx # noqa import onnx # noqa