Skip to content

Fix localizer multi-GPU failure (model.to vs device_map)#404

Merged
KartikP merged 1 commit into
mainfrom
fix-localizer-device-map
Jun 3, 2026
Merged

Fix localizer multi-GPU failure (model.to vs device_map)#404
KartikP merged 1 commit into
mainfrom
fix-localizer-device-map

skip localizer model.to(device) when model is device_map-sharded

86a4c70
Select commit
Loading
Failed to load commit list.

Select a check to view from the sidebar