smg478 commited on
Commit
bb0a5da
1 Parent(s): c2ced98

Comply with transformer 4.46.1

Browse files
Files changed (1) hide show
  1. modeling_h2ovl_chat.py +0 -1
modeling_h2ovl_chat.py CHANGED
@@ -343,7 +343,6 @@ class H2OVLChatModel(PreTrainedModel):
343
  attention_mask=attention_mask,
344
  generation_config=generation_config,
345
  output_hidden_states=output_hidden_states,
346
- return_dict=return_dict,
347
  use_cache=True,
348
  **generate_kwargs,
349
  )
 
343
  attention_mask=attention_mask,
344
  generation_config=generation_config,
345
  output_hidden_states=output_hidden_states,
 
346
  use_cache=True,
347
  **generate_kwargs,
348
  )