Checkpoint dimensionality error: mismation tensor dimensions
#2
by
PsVenom
- opened
So I was trying to run the model using the transformer library and got the following error
ValueError: Flax checkpoint seems to be incorrect. Weight ('final_logits_bias',) was expected to be of shape torch.Size([1, 50264]), but is (1, 16385).
How to run the model? Is there any reference?