Suraj Patil 3d607df8f4 fix loading flax bf16 weights in pt (#14369) ...

* fix loading flax bf16 weights in pt

* fix clip test

* fix t5 test

* add logging statement

* Update src/transformers/modeling_flax_pytorch_utils.py

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>

* switch back to native any

* fix check for bf16 weights

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>

2021-11-11 21:20:49 +05:30

37 KiB

Raw Blame History

View Raw