Commit
·
25dd881
1
Parent(s):
2fc0ee1
Update modeling_phi.py
Browse files- modeling_phi.py +1 -0
modeling_phi.py
CHANGED
|
@@ -382,6 +382,7 @@ class SelfAttention(nn.Module):
|
|
| 382 |
attention = self.drop(attention)
|
| 383 |
|
| 384 |
output = torch.einsum("bhts,bshd->bthd", attention, v)
|
|
|
|
| 385 |
|
| 386 |
return output
|
| 387 |
|
|
|
|
| 382 |
attention = self.drop(attention)
|
| 383 |
|
| 384 |
output = torch.einsum("bhts,bshd->bthd", attention, v)
|
| 385 |
+
print(output[0][0][0])
|
| 386 |
|
| 387 |
return output
|
| 388 |
|