correct device keyword argument to device_map
device
device_map
Thanks
· Sign up or log in to comment