Open maxjeblick opened 1 day ago
Interesting ! Not sure what changed, I also recently ran into this for another model, maybe there has been a recent update on the processor size disabling autocasting or something ! I'll update, but first want to look into it ! thanks !
When I run the example given in the README.md, I encounter the following error:
RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (CUDABFloat16Type) should be the same
Code is working if I cast
pixel_values
explicitly:Stacktrace: