add flax model
This commit is contained in:
parent
2182e92bc6
commit
10fcb38b50
|
@ -3,6 +3,7 @@
|
||||||
"CLIPModel"
|
"CLIPModel"
|
||||||
],
|
],
|
||||||
"initializer_factor": 1.0,
|
"initializer_factor": 1.0,
|
||||||
|
"logit_scale_init_value": 2.6592,
|
||||||
"model_type": "clip",
|
"model_type": "clip",
|
||||||
"projection_dim": 512,
|
"projection_dim": 512,
|
||||||
"text_config": {
|
"text_config": {
|
||||||
|
@ -23,7 +24,6 @@
|
||||||
"finetuning_task": null,
|
"finetuning_task": null,
|
||||||
"forced_bos_token_id": null,
|
"forced_bos_token_id": null,
|
||||||
"forced_eos_token_id": null,
|
"forced_eos_token_id": null,
|
||||||
"gradient_checkpointing": false,
|
|
||||||
"hidden_act": "quick_gelu",
|
"hidden_act": "quick_gelu",
|
||||||
"hidden_size": 512,
|
"hidden_size": 512,
|
||||||
"id2label": {
|
"id2label": {
|
||||||
|
@ -72,7 +72,7 @@
|
||||||
"top_p": 1.0,
|
"top_p": 1.0,
|
||||||
"torch_dtype": null,
|
"torch_dtype": null,
|
||||||
"torchscript": false,
|
"torchscript": false,
|
||||||
"transformers_version": "4.11.0.dev0",
|
"transformers_version": "4.12.0.dev0",
|
||||||
"use_bfloat16": false,
|
"use_bfloat16": false,
|
||||||
"vocab_size": 49408
|
"vocab_size": 49408
|
||||||
},
|
},
|
||||||
|
@ -97,7 +97,6 @@
|
||||||
"finetuning_task": null,
|
"finetuning_task": null,
|
||||||
"forced_bos_token_id": null,
|
"forced_bos_token_id": null,
|
||||||
"forced_eos_token_id": null,
|
"forced_eos_token_id": null,
|
||||||
"gradient_checkpointing": false,
|
|
||||||
"hidden_act": "quick_gelu",
|
"hidden_act": "quick_gelu",
|
||||||
"hidden_size": 768,
|
"hidden_size": 768,
|
||||||
"id2label": {
|
"id2label": {
|
||||||
|
@ -147,7 +146,7 @@
|
||||||
"top_p": 1.0,
|
"top_p": 1.0,
|
||||||
"torch_dtype": null,
|
"torch_dtype": null,
|
||||||
"torchscript": false,
|
"torchscript": false,
|
||||||
"transformers_version": "4.11.0.dev0",
|
"transformers_version": "4.12.0.dev0",
|
||||||
"use_bfloat16": false
|
"use_bfloat16": false
|
||||||
},
|
},
|
||||||
"vision_config_dict": {
|
"vision_config_dict": {
|
||||||
|
|
Binary file not shown.
Loading…
Reference in New Issue