Pin "paddlepaddle<3.0.0" to resolve export bug (#13550)

2025-03-27 08:55:55 +01:00 · 2025-03-27 08:55:55 +01:00 · 6993b75af2
parent ad450f09c6
commit 6993b75af2
1 changed files with 1 additions and 1 deletions
--- 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