diffusers
9918d13e - fix(training_utils): wrap device in list for DiffusionPipeline (#12178)

Commit
81 days ago
fix(training_utils): wrap device in list for DiffusionPipeline (#12178) - Modify offload_models function to handle DiffusionPipeline correctly - Ensure compatibility with both single and multiple module inputs Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
Author
Parents
Loading