Fix KeyError in __init__ when vision_config is empty (Transformers v5 compatibility)

#12
by KBayoud - opened
Files changed (1) hide show
  1. configuration_internvl_chat.py +2 -2
configuration_internvl_chat.py CHANGED
@@ -47,10 +47,10 @@ class InternVLChatConfig(PretrainedConfig):
47
  llm_config = {}
48
  logger.info('llm_config is None. Initializing the LlamaConfig config with default values (`LlamaConfig`).')
49
 
50
- if vision_config and vision_config['model_type']=='intern_vit_patch':
51
  self.vision_config = InternVisionPatchConfig(**vision_config)
52
  else:
53
- raise ValueError('Unsupported vision model type: {}'.format(vision_config['model_type']))
54
  if llm_config['architectures'][0] == 'LlamaForCausalLM':
55
  self.llm_config = LlamaConfig(**llm_config)
56
  elif llm_config['architectures'][0] == 'InternLM2ForCausalLM':
 
47
  llm_config = {}
48
  logger.info('llm_config is None. Initializing the LlamaConfig config with default values (`LlamaConfig`).')
49
 
50
+ if vision_config and vision_config.get('model_type')=='intern_vit_patch':
51
  self.vision_config = InternVisionPatchConfig(**vision_config)
52
  else:
53
+ self.vision_config = None
54
  if llm_config['architectures'][0] == 'LlamaForCausalLM':
55
  self.llm_config = LlamaConfig(**llm_config)
56
  elif llm_config['architectures'][0] == 'InternLM2ForCausalLM':