diff --git a/config.json b/config.json
index 6df928f..aa90347 100644
--- a/config.json
+++ b/config.json
@@ -65,16 +65,16 @@
   "summary_proj_to_labels": true,
   "summary_type": "cls_index",
   "summary_use_proj": true,
-  "transformers_version": "4.5.0.dev0",
-  "use_cache": true,
-  "vocab_size": 50257,
-  "window_size": 256,
-  "tokenizer_class": "GPT2Tokenizer",
   "task_specific_params": {
     "text-generation": {
       "do_sample": true,
-      "temperature": 0.9,
-      "max_length": 50
+      "max_length": 50,
+      "temperature": 0.9
     }
-  }
+  },
+  "tokenizer_class": "GPT2Tokenizer",
+  "transformers_version": "4.9.0.dev0",
+  "use_cache": true,
+  "vocab_size": 50257,
+  "window_size": 256
 }
diff --git a/flax_model.msgpack b/flax_model.msgpack
new file mode 100644
index 0000000..2f052d7
--- /dev/null
+++ b/flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f3734b75e71878322f27d6d834b1c72eb4950e21ed418d2278aeb6a178f33e14
+size 10605245174