valhalla commited on
Commit
10fcb38
1 Parent(s): 2182e92

add flax model

Browse files
Files changed (2) hide show
  1. config.json +3 -4
  2. 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.11.0.dev0",
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.11.0.dev0",
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