add flax model
Browse files- config.json +3 -4
- flax_model.msgpack +3 -0
config.json
CHANGED
|
@@ -3,6 +3,7 @@
|
|
| 3 |
"CLIPModel"
|
| 4 |
],
|
| 5 |
"initializer_factor": 1.0,
|
|
|
|
| 6 |
"model_type": "clip",
|
| 7 |
"projection_dim": 512,
|
| 8 |
"text_config": {
|
|
@@ -23,7 +24,6 @@
|
|
| 23 |
"finetuning_task": null,
|
| 24 |
"forced_bos_token_id": null,
|
| 25 |
"forced_eos_token_id": null,
|
| 26 |
-
"gradient_checkpointing": false,
|
| 27 |
"hidden_act": "quick_gelu",
|
| 28 |
"hidden_size": 512,
|
| 29 |
"id2label": {
|
|
@@ -72,7 +72,7 @@
|
|
| 72 |
"top_p": 1.0,
|
| 73 |
"torch_dtype": null,
|
| 74 |
"torchscript": false,
|
| 75 |
-
"transformers_version": "4.
|
| 76 |
"use_bfloat16": false,
|
| 77 |
"vocab_size": 49408
|
| 78 |
},
|
|
@@ -97,7 +97,6 @@
|
|
| 97 |
"finetuning_task": null,
|
| 98 |
"forced_bos_token_id": null,
|
| 99 |
"forced_eos_token_id": null,
|
| 100 |
-
"gradient_checkpointing": false,
|
| 101 |
"hidden_act": "quick_gelu",
|
| 102 |
"hidden_size": 768,
|
| 103 |
"id2label": {
|
|
@@ -147,7 +146,7 @@
|
|
| 147 |
"top_p": 1.0,
|
| 148 |
"torch_dtype": null,
|
| 149 |
"torchscript": false,
|
| 150 |
-
"transformers_version": "4.
|
| 151 |
"use_bfloat16": false
|
| 152 |
},
|
| 153 |
"vision_config_dict": {
|
|
|
|
| 3 |
"CLIPModel"
|
| 4 |
],
|
| 5 |
"initializer_factor": 1.0,
|
| 6 |
+
"logit_scale_init_value": 2.6592,
|
| 7 |
"model_type": "clip",
|
| 8 |
"projection_dim": 512,
|
| 9 |
"text_config": {
|
|
|
|
| 24 |
"finetuning_task": null,
|
| 25 |
"forced_bos_token_id": null,
|
| 26 |
"forced_eos_token_id": null,
|
|
|
|
| 27 |
"hidden_act": "quick_gelu",
|
| 28 |
"hidden_size": 512,
|
| 29 |
"id2label": {
|
|
|
|
| 72 |
"top_p": 1.0,
|
| 73 |
"torch_dtype": null,
|
| 74 |
"torchscript": false,
|
| 75 |
+
"transformers_version": "4.12.0.dev0",
|
| 76 |
"use_bfloat16": false,
|
| 77 |
"vocab_size": 49408
|
| 78 |
},
|
|
|
|
| 97 |
"finetuning_task": null,
|
| 98 |
"forced_bos_token_id": null,
|
| 99 |
"forced_eos_token_id": null,
|
|
|
|
| 100 |
"hidden_act": "quick_gelu",
|
| 101 |
"hidden_size": 768,
|
| 102 |
"id2label": {
|
|
|
|
| 146 |
"top_p": 1.0,
|
| 147 |
"torch_dtype": null,
|
| 148 |
"torchscript": false,
|
| 149 |
+
"transformers_version": "4.12.0.dev0",
|
| 150 |
"use_bfloat16": false
|
| 151 |
},
|
| 152 |
"vision_config_dict": {
|
flax_model.msgpack
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:eb01703b5ecb458909b5dadbbadd929f7096782dda1f879273bbd38a2c14fc66
|
| 3 |
+
size 598496700
|