correct device keyword argument to device_map

Meta Llama org

Thanks

ArthurZ changed pull request status to merged
Your need to confirm your account before you can post a new comment.

Sign up or log in to comment