riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 02:31:11 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Fix link in pipeline device map (#20517)

Browse Source 
* fix link in pipeline device map

* oops this is the correct link

* make style
This commit is contained in:
Steven Liu 2022-12-01 09:58:44 -08:00 committed by GitHub
parent 504ae9181c
commit 5011efbec8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/transformers/pipelines/__init__.py
View File

@ -558,8 +558,9 @@ def pipeline(
pipeline will be allocated. pipeline will be allocated.
device_map (`str` or `Dict[str, Union[int, str, torch.device]`, *optional*): device_map (`str` or `Dict[str, Union[int, str, torch.device]`, *optional*):
Sent directly as `model_kwargs` (just a simpler shortcut). When `accelerate` library is present, set Sent directly as `model_kwargs` (just a simpler shortcut). When `accelerate` library is present, set
`device_map="auto"` to compute the most optimized `device_map` automatically. [More `device_map="auto"` to compute the most optimized `device_map` automatically (see
information](https://huggingface.co/docs/accelerate/main/en/big_modeling#accelerate.cpu_offload) [here](https://huggingface.co/docs/accelerate/main/en/package_reference/big_modeling#accelerate.cpu_offload)
for more information).
<Tip warning={true}> <Tip warning={true}>