Correctly set device_map

czczup changed pull request status to merged

Sign up or log in to comment