search

VectorSpaceLab / OmniGen

OmniGen: Unified Image Generation. https://arxiv.org/pdf/2409.11340
MIT License
2.53k stars 187 forks source link

allow custom device specification inside OmniGenPipeline #85

Closed bghira closed 1 week ago

bghira commented 1 week ago

this allows using the non-default device. or something not yet supported by this class.

staoxiao commented 1 week ago

Thanks!