Skip to content

GitLab

Unverified Commit c82b38a3 authored by amyeroberts's avatar amyeroberts Committed by GitHub
Browse files

Fix pipeline logger.warning_once bug (#30195) 

Fix warning bug
parent 2c66600c
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
src/transformers/pipelines/base.py
View file @ c82b38a3
...@@ -1181,7 +1181,6 @@ class Pipeline(_ScikitCompat): ...@@ -1181,7 +1181,6 @@ class Pipeline(_ScikitCompat):
logger.warning_once( logger.warning_once(
"You seem to be using the pipelines sequentially on GPU. In order to maximize efficiency please use a" "You seem to be using the pipelines sequentially on GPU. In order to maximize efficiency please use a"
" dataset", " dataset",
UserWarning,
) )
is_dataset = Dataset is not None and isinstance(inputs, Dataset) is_dataset = Dataset is not None and isinstance(inputs, Dataset)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment