tt1225
/

openvla-7b-devel

Image-Text-to-Text

Model card Files Files and versions

tt1225 commited on Apr 29

Commit

8f4ed79

·

verified ·

1 Parent(s): 262891d

Update modeling_prismatic.py

Files changed (1) hide show

modeling_prismatic.py +1 -1

modeling_prismatic.py CHANGED Viewed

@@ -438,7 +438,7 @@ class PrismaticForConditionalGeneration(PrismaticPreTrainedModel):
             return language_model_output
-        print("loss", language_model_output.loss.shape)
         print("logits", language_model_output.logits.shape)
         print("hidden_states", language_model_output.hidden_states.shape)
         print("attentions", language_model_output.attentions.shape)

             return language_model_output
+        # print("loss", language_model_output.loss.shape)
         print("logits", language_model_output.logits.shape)
         print("hidden_states", language_model_output.hidden_states.shape)
         print("attentions", language_model_output.attentions.shape)