nathanrchn commited on
Commit
25dd881
·
1 Parent(s): 2fc0ee1

Update modeling_phi.py

Browse files
Files changed (1) hide show
  1. 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