sgugger commited on
Commit
e73acbb
1 Parent(s): e7d09e2

Fix weights by putting the right value in `lm_head.weight`

Browse files

There was probably a bug in the initial conversion script that created those models, as the weights they have have a
different value for `lm_head.weight` and `model.decoder.embed_tokens.weight`. Those models are tied though.

This was not a problem until now as the model was tied after the load and the (wrong) value of `lm_head.weight` was
replaced by the value of `model.decoder.embed_tokens.weight`. This does not work any more if we tie the weights before
the load however, as the value picked might be the one from `lm_head.weight` depending on how the models are tied.
As far as I can see, the model stop generating properly on Transformers main.

This should fix the bug without any side effect.

Files changed (1) hide show
  1. pytorch_model.bin +2 -2
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:eb6360d6fd4f5e2ba6a397474ec3416ac3286e0b7509e81d400b0dd7b01cd4dc
3
- size 153146267
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4f4fd19ef323cf6b419cd21b43f3b13d4a1722f544b9240ad5bfd5ad71e7c4bb
3
+ size 153143557