aws / aws/sagemaker-huggingface-inference-toolkit

No support for multi-GPU

Open
#106 3 comments 0 reactions 0 assignees View on GitHub
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

Open the contributing 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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.