Fix for 'got multiple values for keyword argument 'return_dict'
#1
by
Blackroot
- opened
- 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 |
)
|