tt1225
/

openvla-7b-devel

Image-Text-to-Text

Model card Files Files and versions

tt1225 commited on Apr 29

Commit

8c7cc62

·

verified ·

1 Parent(s): 6db9b69

Update modeling_prismatic.py

Files changed (1) hide show

modeling_prismatic.py +3 -1

modeling_prismatic.py CHANGED Viewed

@@ -443,7 +443,9 @@ class PrismaticForConditionalGeneration(PrismaticPreTrainedModel):
         print("hidden_states")
         # for i in language_model_output.hidden_states:
         #     print(i.shape)
-        print("attentions", language_model_output.attentions.shape)
         return PrismaticCausalLMOutputWithPast(
             loss=language_model_output.loss,

         print("hidden_states")
         # for i in language_model_output.hidden_states:
         #     print(i.shape)
+        print("attentions")
+        for i in language_model_output.attentions:
+            print(i.shape)
         return PrismaticCausalLMOutputWithPast(
             loss=language_model_output.loss,