From 1e1707aeb4ec7c9f254822b37e4a7ad960db3d98 Mon Sep 17 00:00:00 2001 From: An Yang Date: Wed, 23 Nov 2022 15:23:18 +0000 Subject: [PATCH] Update config.json --- config.json | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/config.json b/config.json index c25912c..bcdd158 100644 --- a/config.json +++ b/config.json @@ -8,7 +8,7 @@ "projection_dim": 512, "text_config": { "architectures": [ - "BertModel" + "ChineseCLIPTextModel" ], "attention_probs_dropout_prob": 0.1, "bos_token_id": 0, @@ -21,7 +21,7 @@ "intermediate_size": 3072, "layer_norm_eps": 1e-12, "max_position_embeddings": 512, - "model_type": "bert", + "model_type": "chinese-clip_text_model", "num_attention_heads": 12, "num_hidden_layers": 12, "output_past": true, @@ -75,7 +75,7 @@ "length_penalty": 1.0, "max_length": 20, "min_length": 0, - "model_type": "clip_vision_model", + "model_type": "chinese-clip_vision_model", "no_repeat_ngram_size": 0, "num_attention_heads": 12, "num_beam_groups": 1,