Duplicated from akdeniz27/pix2struct-DocVQA
Cast the model in bfloat16 and put the model and input on the correct device
bfloat16
Thanks, Younes!
· Sign up or log in to comment