Fix pipeline logger.warning_once bug (#30195)

Fix warning bug

Fix pipeline logger.warning_once bug (#30195)
Fix warning bug
c82b38a3 · amyeroberts · GitHub · 2c66600c · c82b38a3
Unverified Commit c82b38a3 authored Apr 12, 2024 by amyeroberts Committed by GitHub Apr 12, 2024
Show whitespace changes
Inline Side-by-side

Showing with 0 additions and 1 deletion

src/transformers/pipelines/base.py src/transformers/pipelines/base.py +0 -1

No files found.
--- a/src/transformers/pipelines/base.py
+++ b/src/transformers/pipelines/base.py
@@ -1181,7 +1181,6 @@ class Pipeline(_ScikitCompat):
            logger.warning_once(
                "You seem to be using the pipelines sequentially on GPU. In order to maximize efficiency please use a"
                " dataset",
-                UserWarning,
            )

        is_dataset = Dataset is not None and isinstance(inputs, Dataset)