Unverified Commit df6f36a1 authored by Hz, Ji's avatar Hz, Ji Committed by GitHub
Browse files

deprecate function `get_default_device` in `tools/base.py` (#26774)



* get default device through `PartialState().default_device` as is has
been officially released

* apply code review suggestion

* apply code review suggestion
Co-authored-by: default avatarZach Mueller <muellerzr@gmail.com>

---------
Co-authored-by: default avatarZach Mueller <muellerzr@gmail.com>
parent 8211c59b
......@@ -599,6 +599,10 @@ def launch_gradio_demo(tool_class: Tool):
# TODO: Migrate to Accelerate for this once `PartialState.default_device` makes its way into a release.
def get_default_device():
logger.warning(
"`get_default_device` is deprecated and will be replaced with `accelerate`'s `PartialState().default_device` "
"in version 4.36 of 🤗 Transformers. "
)
if not is_torch_available():
raise ImportError("Please install torch in order to use this tool.")
......
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