Unverified Commit aac8ee42 authored by Hans Wang's avatar Hans Wang Committed by GitHub
Browse files

Fix the error caused by incorrect use of logger in pipeline (#31565)

parent c54a8ca4
......@@ -850,7 +850,7 @@ class Pipeline(_ScikitCompat, PushToHubMixin):
or is_torch_xpu_available(check_device=True)
or is_torch_mps_available()
):
logging.warning(
logger.warning(
"Hardware accelerator e.g. GPU is available in the environment, but no `device` argument"
" is passed to the `Pipeline` object. Model will be on CPU."
)
......
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