Skip to content

Unified attention#199

Open
israfelsr wants to merge 2 commits intomasterfrom
unified-attention
Open

Unified attention#199
israfelsr wants to merge 2 commits intomasterfrom
unified-attention

Conversation

@israfelsr
Copy link
Contributor

Created function that transfer weights of pretrained models to the new CasualAttention module.
Tested with some prompts and working perfectly 🤗.
Will add it to student branch.

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.

1 participant