tt1225
/

openvla-7b-devel

Image-Text-to-Text

Model card Files Files and versions

tt1225 commited on Apr 29

Commit

a62d3bd

·

verified ·

1 Parent(s): 853de96

Update modeling_prismatic.py

Files changed (1) hide show

modeling_prismatic.py +2 -0

modeling_prismatic.py CHANGED Viewed

@@ -485,6 +485,8 @@ class PrismaticForConditionalGeneration(PrismaticPreTrainedModel):
             }
         )
         print("Prepare")
         print(model_inputs.keys())

             }
         )
+        model_inputs["output_attentions"] = True
         print("Prepare")
         print(model_inputs.keys())