how to use device_map in diffusers.stablediffusionpipeline.from_pretrained #8646
Open
Description
❓ Questions and Help
https://huggingface.co/docs/accelerate/main/en/usage_guides/big_modeling#designing-a-device-map
i try device_map=auto. but it seems not recongize the xla device tpu all goto cpu. please help and provide the minimum working example. @JackCaoG
Metadata
Assignees
Labels
No labels