Release InternVL 2.0 pretrained models
Browse files
InternVL2-1B-Pretrain/config.json
CHANGED
@@ -81,7 +81,7 @@
|
|
81 |
"temperature": 1.0,
|
82 |
"tf_legacy_loss": false,
|
83 |
"tie_encoder_decoder": false,
|
84 |
-
"tie_word_embeddings":
|
85 |
"tokenizer_class": null,
|
86 |
"top_k": 50,
|
87 |
"top_p": 1.0,
|
|
|
81 |
"temperature": 1.0,
|
82 |
"tf_legacy_loss": false,
|
83 |
"tie_encoder_decoder": false,
|
84 |
+
"tie_word_embeddings": false,
|
85 |
"tokenizer_class": null,
|
86 |
"top_k": 50,
|
87 |
"top_p": 1.0,
|
InternVL2-1B-Pretrain/configuration_internvl_chat.py
CHANGED
@@ -38,11 +38,11 @@ class InternVLChatConfig(PretrainedConfig):
|
|
38 |
super().__init__(**kwargs)
|
39 |
|
40 |
if vision_config is None:
|
41 |
-
vision_config = {}
|
42 |
logger.info('vision_config is None. Initializing the InternVisionConfig with default values.')
|
43 |
|
44 |
if llm_config is None:
|
45 |
-
llm_config = {}
|
46 |
logger.info('llm_config is None. Initializing the LlamaConfig config with default values (`LlamaConfig`).')
|
47 |
|
48 |
self.vision_config = InternVisionConfig(**vision_config)
|
|
|
38 |
super().__init__(**kwargs)
|
39 |
|
40 |
if vision_config is None:
|
41 |
+
vision_config = {'architectures': ['InternVisionModel']}
|
42 |
logger.info('vision_config is None. Initializing the InternVisionConfig with default values.')
|
43 |
|
44 |
if llm_config is None:
|
45 |
+
llm_config = {'architectures': ['Qwen2ForCausalLM']}
|
46 |
logger.info('llm_config is None. Initializing the LlamaConfig config with default values (`LlamaConfig`).')
|
47 |
|
48 |
self.vision_config = InternVisionConfig(**vision_config)
|