add flax model

This commit is contained in:
patil-suraj 2021-10-04 12:07:56 +02:00
parent 2182e92bc6
commit 10fcb38b50
2 changed files with 6 additions and 4 deletions

View File

@ -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": {

BIN
flax_model.msgpack (Stored with Git LFS) Normal file

Binary file not shown.