Invalid scalar type

When training is done on gpt-j-6B using ray Air , invalid scalar type error is occured . What is the reason for this? How to overcome this?

Sample example performed which is in the website .

Hey @apple1234 can you share the error message with us?