Update configuration_h2ovl_chat.py
#7
by
mlandry22
- opened
configuration_h2ovl_chat.py
CHANGED
|
@@ -44,7 +44,7 @@ class H2OVLChatConfig(PretrainedConfig):
|
|
| 44 |
logger.info('vision_config is None. Initializing the InternVisionConfig with default values.')
|
| 45 |
|
| 46 |
if llm_config is None:
|
| 47 |
-
llm_config = {'architectures': ['
|
| 48 |
logger.info('llm_config is None. Initializing the LlamaConfig config with default values (`LlamaConfig`).')
|
| 49 |
|
| 50 |
self.vision_config = InternVisionConfig(**vision_config)
|
|
|
|
| 44 |
logger.info('vision_config is None. Initializing the InternVisionConfig with default values.')
|
| 45 |
|
| 46 |
if llm_config is None:
|
| 47 |
+
llm_config = {'architectures': ['MistralForCausalLM']}
|
| 48 |
logger.info('llm_config is None. Initializing the LlamaConfig config with default values (`LlamaConfig`).')
|
| 49 |
|
| 50 |
self.vision_config = InternVisionConfig(**vision_config)
|