From 10fcb38b5006b9e1f302d0b05ad5b2201dc844da Mon Sep 17 00:00:00 2001 From: patil-suraj Date: Mon, 4 Oct 2021 12:07:56 +0200 Subject: [PATCH] add flax model --- config.json | 7 +++---- flax_model.msgpack | 3 +++ 2 files changed, 6 insertions(+), 4 deletions(-) create mode 100644 flax_model.msgpack diff --git a/config.json b/config.json index 2498330..eade54b 100644 --- a/config.json +++ b/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": { diff --git a/flax_model.msgpack b/flax_model.msgpack new file mode 100644 index 0000000..3c70a8f --- /dev/null +++ b/flax_model.msgpack @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:eb01703b5ecb458909b5dadbbadd929f7096782dda1f879273bbd38a2c14fc66 +size 598496700