4 Commits

Author	SHA1	Message	Date
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
Patrick von Platen	60e448c87e	[Flax] Correct pt to flax conversion if from base to head (#13006) ... * finish PR * add tests * correct tests * finish * correct other flax tests * better naming * correct naming * finish * apply sylvains suggestions	2021-08-05 18:38:50 +02:00
Patrick von Platen	cee2d2135f	[Flax Generation] Correct inconsistencies PyTorch/Flax (#12662) ... * fix_torch_device_generate_test * remove @ * correct greedy search * save intertmed * add final logits bias * correct * up * add more tests * fix another bug * finish tests * finish marian tests * up Co-authored-by: Patrick von Platen <patrick@huggingface.co>	2021-07-13 18:53:30 +01:00
Vasudev Gupta	e98233dde1	Flax T5 (#12150) ... * copy pytorch-t5 * init * boom boom * forward pass same * make generation work * add more tests * make test work * finish normal tests * make fix-copies * finish quality * correct slow example * correct slow test * version table * upload models * Update tests/test_modeling_flax_t5.py * correct incorrectly deleted line Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> Co-authored-by: Patrick von Platen <patrick@huggingface.co>	2021-06-23 13:13:32 +01:00