optional) —
A map that specifies where each submodule should go. It doesn’t need to be defined for each
parameter/buffer name; once a given module name is inside, every submodule of it will be sent to the
same device.
Set device_map="auto" to have 🤗 Accelerate automatically compute the most optimized device_map. For
more information about each option see designing a device
map.   max_memory (Dict, optional) —
A dictionary device identifier for 