BBing
/
mindspore
forked from mindspore/mindspore
0
0
Fork 2
Code Issues Pull Requests 1 Packages Projects Releases Wiki Activity

Update attn.py

This commit is contained in:
BBing 2024-10-15 21:54:12 +08:00
parent 26b51412c9
commit ab462cb07d
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
predict/models/attn.py
View File

@ -31,4 +31,6 @@ class FullAttention(nn.Cell):
if self.output_attention:
return V, A
else:
return V, None
return V, None
class ProbAttention(nn.Cell):