Correctly set the device_map in the example

czczup changed pull request status to merged

Sign up or log in to comment