add flax model

config.json

@@ -3,6 +3,7 @@
     "CLIPModel"
   ],
   "initializer_factor": 1.0,
+  "logit_scale_init_value": 2.6592,
   "model_type": "clip",
   "projection_dim": 512,
   "text_config": {
@@ -23,7 +24,6 @@
     "finetuning_task": null,
     "forced_bos_token_id": null,
     "forced_eos_token_id": null,
-    "gradient_checkpointing": false,
     "hidden_act": "quick_gelu",
     "hidden_size": 512,
     "id2label": {
@@ -72,7 +72,7 @@
     "top_p": 1.0,
     "torch_dtype": null,
     "torchscript": false,
-    "transformers_version": "4.11.0.dev0",
+    "transformers_version": "4.12.0.dev0",
     "use_bfloat16": false,
     "vocab_size": 49408
   },
@@ -97,7 +97,6 @@
     "finetuning_task": null,
     "forced_bos_token_id": null,
     "forced_eos_token_id": null,
-    "gradient_checkpointing": false,
     "hidden_act": "quick_gelu",
     "hidden_size": 768,
     "id2label": {
@@ -147,7 +146,7 @@
     "top_p": 1.0,
     "torch_dtype": null,
     "torchscript": false,
-    "transformers_version": "4.11.0.dev0",
+    "transformers_version": "4.12.0.dev0",
     "use_bfloat16": false
   },
   "vision_config_dict": {