transformers
e9a5e32b - [blip-2] Fix dtype mismatch when keep in fp32 (#37068)

Commit

276 days ago

[blip-2] Fix dtype mismatch when keep in fp32 (#37068) * fix fp32 BLIP2 * no need to reorder that * check for `Noneness` as well before casting dtype

Author

zucchini-nlp

Committer

ArthurZucker

Parents