"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "3ec40299c18c370011a39e0576faf1e90c0a8164"
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