Invalid scalar type

Ray AIR (Data, Train, Tune, Serve)
1

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 .

2

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