Mentors4EDU's picture
Declare _no_split_modules so device_map='auto' can shard the model across GPUs
ec3e3c1 verified