me here.


device_map (str or Dict[str, Union[int, str, torch.device]], optional) —
A map that specifies where each submodule should go. It doesn’t need to be refined to each
parameter/buffer name, once a given module name is inside, every submodule of it will be sent to the
same device.
To have Accelerate compute the most optimized device_map automatically, set device_map="auto". For
more information about each option see designing a device
ma