tt1225
/

openvla-7b-devel

Image-Text-to-Text

Model card Files Files and versions

tt1225 commited on Apr 29

Commit

97d9790

·

verified ·

1 Parent(s): 5a1d866

Update modeling_prismatic.py

Files changed (1) hide show

modeling_prismatic.py +1 -1

modeling_prismatic.py CHANGED Viewed

@@ -408,7 +408,7 @@ class PrismaticForConditionalGeneration(PrismaticPreTrainedModel):
                 use_cache=use_cache,
                 output_attentions=True,
                 output_hidden_states=output_hidden_states,
-                return_dict=True,
             )
         # === Otherwise =>> Assume Invalid! ===

                 use_cache=use_cache,
                 output_attentions=True,
                 output_hidden_states=output_hidden_states,
+                return_dict=return_dict,
             )
         # === Otherwise =>> Assume Invalid! ===