transformers
[Flax `.from_pretrained`] Raise a warning if model weights are not in float32
#16762

Merged

[Flax `.from_pretrained`] Raise a warning if model weights are not in float32 #16762

patil-suraj merged 3 commits into huggingface:main from sanchit-gandhi:flax-from-pretrained

[Flax] Raise a warning if model weights are not in float32

09d38b3e

sanchit-gandhi changed the title ~~[Flax .from_pretrained] Raise a warning if model weights are not in float32~~ [Flax `.from_pretrained`] Raise a warning if model weights are not in float32 4 years ago

sanchit-gandhi requested a review from

patrickvonplaten 4 years ago

sanchit-gandhi requested a review from

patil-suraj 4 years ago

patil-suraj approved these changes on 2022-04-13

apply suggestions and few small changes

9eac5ecd

reorder wording for better readability

368e61de

patil-suraj commented on 2022-04-14

patil-suraj merged d8269eb4 into main 4 years ago

sanchit-gandhi deleted the flax-from-pretrained branch 3 years ago

Reviewers

patil-suraj

patrickvonplaten

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

transformers [Flax `.from_pretrained`] Raise a warning if model weights are not in float32 #16762 Merged

[Flax `.from_pretrained`] Raise a warning if model weights are not in float32 #16762

transformers
[Flax `.from_pretrained`] Raise a warning if model weights are not in float32
#16762

Merged