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