correct checkpoints see: https://github.com/facebookresearch/metaseq/pull/164

config.json

@@ -20,8 +20,8 @@
   "num_hidden_layers": 12,
   "pad_token_id": 1,
   "prefix": "</s>",
-  "torch_dtype": "float32",
-  "transformers_version": "4.20.0.dev0",
+  "torch_dtype": "float16",
+  "transformers_version": "4.21.0.dev0",
   "use_cache": true,
   "vocab_size": 50272,
   "word_embed_proj_dim": 768

flax_model.msgpack

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f90cb1ba90690ebb648d58f6855b18052bd0611383316e720b4210d30a547e22
-size 500957820
+oid sha256:61c60ec52ed43038fff0fbbd68b080c94b0d94b4c8458dbd65965f9b17631c89
+size 250485441

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ae15ed1fe2c874c61432747b9d3d23654e9ca29c650e0c8310eff25c8709599d
-size 250535293
+oid sha256:2d74da6615135c58cf3cf9ad4cb11e7c613ff9e55fe658a47ab83b6c8d1174a9
+size 250540281

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4287e364c96263abb962fc0c8d85a05efdcdaaa32baabfd8a87e4e791b2c44db
-size 250704264
+oid sha256:db7090b0c8b34dd957a7e0656c718f978f9203cc874018f37dda44108be5970a
+size 250709016