required by the save_pretrained function in transformers>=4.45

#4
by guanshuo - opened
Files changed (1) hide show
  1. configuration_h2ovl_chat.py +1 -1
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 = {}
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': ['LlamaForCausalLM']}
48
  logger.info('llm_config is None. Initializing the LlamaConfig config with default values (`LlamaConfig`).')
49
 
50
  self.vision_config = InternVisionConfig(**vision_config)