Skip to content
This repository has been archived by the owner on Sep 17, 2024. It is now read-only.

Commit

Permalink
Fix the error caused by incorrect use of logger in pipeline (huggingf…
Browse files Browse the repository at this point in the history
  • Loading branch information
lanyun1103 authored Jun 24, 2024
1 parent c54a8ca commit aac8ee4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/transformers/pipelines/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -850,7 +850,7 @@ def __init__(
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."
)
Expand Down

0 comments on commit aac8ee4

Please sign in to comment.