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