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,