Skip to content

Specify which device to run on#603

Open
delalamo wants to merge 5 commits intojwohlwend:mainfrom
delalamo:specify_device
Open

Specify which device to run on#603
delalamo wants to merge 5 commits intojwohlwend:mainfrom
delalamo:specify_device

Conversation

@delalamo
Copy link

Currently Boltz does not provide a way to specify which device to run on when multiple GPUs are available. This backward-compatible pull request addresses that by allowing the user to pass either an int, which is currently the default and communicates how many devices to run, or a comma-separated list, which specifies which devices to run on. A single device can be specified by passing --devices 1,.

@gcorso
Copy link
Collaborator

gcorso commented Oct 25, 2025

Hi Diego, thank you for suggesting the change! Could you add some documentation to the inference docs to explain the updated usage of the parameter?

@delalamo
Copy link
Author

Sounds good, adding this now to prediction.md, let me know if it should go anywhere else.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants