diffusers

Fix loading sharded checkpoints when we have variants

#9061

Merged