Skip to content

Commit

Permalink
Merge branch 'mask' of github.com:chengchingwen/NeuralAttentionlib.jl…
Browse files Browse the repository at this point in the history
… into mask
  • Loading branch information
chengchingwen committed May 28, 2024
2 parents b4f03c5 + 3af9063 commit 4312174
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 2 deletions.
1 change: 0 additions & 1 deletion src/mask/wrapper.jl
Original file line number Diff line number Diff line change
Expand Up @@ -174,7 +174,6 @@ function bi_destsize(destsize, is_q)
i, j, J... = destsize
end
return (1, ifelse(is_q, j, i), J...)
# return is_q ? (1, j, J...) : (1, i, J...)
end

Base.@propagate_inbounds function maskgetindex(destsize::Dims, m::BiSequenceMask, i::Integer, j::Integer, J::Integer...)
Expand Down
1 change: 0 additions & 1 deletion src/module/mask.jl
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,6 @@ Return the [`Indexer`](@ref) of `m` and check if the mask `m` can be applied to
"""
GetIndexer


"""
AbstractDatalessMask <: AbstractAttenMask
Expand Down

0 comments on commit 4312174

Please sign in to comment.