Fixed minor typo and added multi GPU case

#2

Very nice implementation. You had a little typo in transformers"y" and I added the special case of using multiple GPUs and getting the specified error.

bakrianoo changed pull request status to merged

Thank you :)

Sign up or log in to comment