aws / aws/sagemaker-huggingface-inference-toolkit
No support for multi-GPU
Open
- Dominant language
- Python
- Stars
- 270
- Forks
- 60
- PR merge metrics
- No merged PRs in 30d
Description
It seems that it's not possible to run models using multiple gpus, e.g. by passing `device_map="auto"` to pipelines.
Is there any way to work around this limitation?
Contributor guide
Research direction
Start by tracing how pipelines are configured and how the `device_map="auto"` argument is handled in the inference toolkit. Reproduce the reported multi-GPU limitation with a model and pipeline, then document the supported behavior or define the changes and validation needed for multi-GPU execution.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- machine-learning
- Issue type
- Feature
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100