Specify which device to run on
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,.
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?
Sounds good, adding this now to prediction.md, let me know if it should go anywhere else.