add flax model

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
 }