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)
|