Update modeling_prismatic.py
Browse files- modeling_prismatic.py +2 -0
modeling_prismatic.py
CHANGED
|
@@ -429,6 +429,8 @@ class PrismaticForConditionalGeneration(PrismaticPreTrainedModel):
|
|
| 429 |
f"=> `use_cache` = {use_cache}"
|
| 430 |
)
|
| 431 |
|
|
|
|
|
|
|
| 432 |
# Unpack `language_model_output` and return PrismaticCausalLMOutputWithPast (or tuple if not `return_dict`)
|
| 433 |
if not return_dict:
|
| 434 |
if output_projector_features and (projected_patch_embeddings is not None):
|
|
|
|
| 429 |
f"=> `use_cache` = {use_cache}"
|
| 430 |
)
|
| 431 |
|
| 432 |
+
return_dict = True
|
| 433 |
+
|
| 434 |
# Unpack `language_model_output` and return PrismaticCausalLMOutputWithPast (or tuple if not `return_dict`)
|
| 435 |
if not return_dict:
|
| 436 |
if output_projector_features and (projected_patch_embeddings is not None):
|