transformers

[device_map] Fix device_map computation by correctly adjusting memory available

#44565

Merged