added tf and flax support

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "/tmp/test-ner",
+  "_name_or_path": "ElectraForTokenClassificationNER",
   "architectures": [
     "ElectraForTokenClassification"
   ],
@@ -46,7 +46,7 @@
   "summary_type": "first",
   "summary_use_proj": true,
   "torch_dtype": "float32",
-  "transformers_version": "4.18.0.dev0",
+  "transformers_version": "4.19.0.dev0",
   "type_vocab_size": 2,
   "use_cache": true,
   "vocab_size": 30522

convert_pytorch_to_flax.py

@@ -0,0 +1,3 @@
+from transformers import FlaxAutoModelForTokenClassification
+model = FlaxAutoModelForTokenClassification.from_pretrained("./", from_pt=True)
+model.save_pretrained("./")

flax_model.msgpack

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:582ff1ae9063765443c6f6fed56d2cd82b913025abdd6d37ac66f0b9d4539cb9
+size 435601421

tf_model.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:004becf31e3f33494b1b76ea91193c7b4a40448b0e4a86f358644241a9f41cf8
+size 435882528