Update modeling_prismatic.py
Browse files- modeling_prismatic.py +5 -0
modeling_prismatic.py
CHANGED
|
@@ -438,6 +438,11 @@ class PrismaticForConditionalGeneration(PrismaticPreTrainedModel):
|
|
| 438 |
|
| 439 |
return language_model_output
|
| 440 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 441 |
return PrismaticCausalLMOutputWithPast(
|
| 442 |
loss=language_model_output.loss,
|
| 443 |
logits=language_model_output.logits,
|
|
|
|
| 438 |
|
| 439 |
return language_model_output
|
| 440 |
|
| 441 |
+
print("loss", language_model_output.loss.shape)
|
| 442 |
+
print("logits", language_model_output.logits.shape)
|
| 443 |
+
print("hidden_states", language_model_output.hidden_states.shape)
|
| 444 |
+
print("attentions", language_model_output.attentions.shape)
|
| 445 |
+
|
| 446 |
return PrismaticCausalLMOutputWithPast(
|
| 447 |
loss=language_model_output.loss,
|
| 448 |
logits=language_model_output.logits,
|