Upload folder using huggingface_hub

configuration_sa2va_chat.py

@@ -39,7 +39,6 @@ class Sa2VAChatConfig(PretrainedConfig):
             max_dynamic_patch=6,
             **kwargs):
         super().__init__(**kwargs)
-
         if vision_config is None:
             vision_config = {}
             logger.info('vision_config is None. Initializing the InternVisionConfig with default values.')
@@ -49,6 +48,7 @@ class Sa2VAChatConfig(PretrainedConfig):
             logger.info('llm_config is None. Initializing the LlamaConfig config with default values (`LlamaConfig`).')
 
         self.vision_config = InternVisionConfig(**vision_config)
+
         if llm_config['architectures'][0] == 'LlamaForCausalLM':
             self.llm_config = LlamaConfig(**llm_config)
         elif llm_config['architectures'][0] == 'InternLM2ForCausalLM':

modeling_sa2va_chat.py

@@ -513,7 +513,7 @@ class Sa2VAChatModel(PreTrainedModel):
             attention_mask=attention_mask.to(device),
             generation_config=generation_config,
             output_hidden_states=output_hidden_states,
-            return_dict=return_dict,
+            # return_dict=return_dict,
             use_cache=True,
             **generate_kwargs,
         )