Skip to content

Conversation

@c0g
Copy link

@c0g c0g commented Sep 25, 2025

Pushed this to clean up, will do a few iterations of self review

Tom Nickson added 9 commits September 5, 2025 19:55
- Add useBidirectionalAttention config parameter
- Apply sliding window size adjustment for bidirectional mode
- Implement createBidirectionalSlidingWindowMask function
- Update mask creation logic to support both causal and bidirectional attention
- Based on patches 40694 and 40700 for EmbeddingGemma support
@davidkoski
Copy link
Collaborator

Hi, this needs to be rebased on main and swift-format applied (make sure to use 602.0.0). Thanks!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants