diff --git a/export.py b/export.py index 3ecb353b7..b9dc1681a 100644 --- a/export.py +++ b/export.py @@ -510,7 +510,7 @@ def export_paddle(model, im, file, metadata, prefix=colorstr("PaddlePaddle:")): $ pip install paddlepaddle x2paddle ``` """ - check_requirements(("paddlepaddle", "x2paddle")) + check_requirements(("paddlepaddle<3.0.0", "x2paddle")) import x2paddle from x2paddle.convert import pytorch2paddle diff --git a/models/common.py b/models/common.py index ea893db4b..aa7898275 100644 --- a/models/common.py +++ b/models/common.py @@ -646,7 +646,7 @@ class DetectMultiBackend(nn.Module): raise NotImplementedError("ERROR: YOLOv5 TF.js inference is not supported") elif paddle: # PaddlePaddle LOGGER.info(f"Loading {w} for PaddlePaddle inference...") - check_requirements("paddlepaddle-gpu" if cuda else "paddlepaddle") + check_requirements("paddlepaddle-gpu" if cuda else "paddlepaddle<3.0.0") import paddle.inference as pdi if not Path(w).is_file(): # if not *.pdmodel