h2oai
/

h2ovl-mississippi-800m

Text Generation

feature-extraction

multimodal large language model

Model card Files Files and versions

GuanshuoXu commited on Oct 31, 2024

Commit

ba973d4

·

1 Parent(s): a5d1ed5

solve transformers 4.46 error

Files changed (1) hide show

modelling_h2ovl_chat.py +0 -2

modelling_h2ovl_chat.py CHANGED Viewed

@@ -335,7 +335,6 @@ class H2OVLChatModel(PreTrainedModel):
             attention_mask=attention_mask,
             generation_config=generation_config,
             output_hidden_states=output_hidden_states,
-            return_dict=return_dict,
             use_cache=True,
             **generate_kwargs,
         )
@@ -468,7 +467,6 @@ class H2OVLChatModel(PreTrainedModel):
             attention_mask=attention_mask,
             generation_config=generation_config,
             output_hidden_states=output_hidden_states,
-            return_dict=return_dict,
             use_cache=True,
             **generate_kwargs,
         )

             attention_mask=attention_mask,
             generation_config=generation_config,
             output_hidden_states=output_hidden_states,
             use_cache=True,
             **generate_kwargs,
         )
             attention_mask=attention_mask,
             generation_config=generation_config,
             output_hidden_states=output_hidden_states,
             use_cache=True,
             **generate_kwargs,
         )