correct device keyword argument to device_map

Meta Llama org

Thanks

ArthurZ changed pull request status to merged

Sign up or log in to comment