Fix for 'got multiple values for keyword argument 'return_dict'

#1
Files changed (1) hide show
  1. modeling_sa2va_chat.py +1 -1
modeling_sa2va_chat.py CHANGED
@@ -513,7 +513,7 @@ class Sa2VAChatModel(PreTrainedModel):
513
  attention_mask=attention_mask.to(device),
514
  generation_config=generation_config,
515
  output_hidden_states=output_hidden_states,
516
- return_dict=return_dict,
517
  use_cache=True,
518
  **generate_kwargs,
519
  )
 
513
  attention_mask=attention_mask.to(device),
514
  generation_config=generation_config,
515
  output_hidden_states=output_hidden_states,
516
+ #return_dict=return_dict,
517
  use_cache=True,
518
  **generate_kwargs,
519
  )