Skip to content

[Feature] Add arguments for selecting GPU to run on #1240

Description

@MrDrMcCoy

Feature Summary

Add an equivalent argument to --main-gpu from llama.cpp

Detailed Description

There is presently no way to tell stable-diffusion.cpp which GPU to use if multiple are available. This makes queuing batch jobs across multiple cards unnecessarily difficult. It would be useful to have an equivalent to llama.cpp's --main-gpu along with related headers for wrappers like cyllama to interface with.

Alternatives you considered

Although hardware acceleration frameworks can pick up environment variables like HIP_VISIBLE_DEVICES and CUDA_VISIBLE_DEVICES to select device to run on, this is quite cumbersome and requires elaborate "shell-out" scripting for batch jobs. Having something internal to the application to control device selection would make larger scale usage of stable-diffusion.cpp much more viable.

Additional context

The absence of this feature blocks 3rd-party wrappers from using stable-diffusion.cpp at scale: shakfu/cyllama#11

Activity

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or request

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions