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