-
Notifications
You must be signed in to change notification settings - Fork 1k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add tensor parallelism for RWKV #1237
Open
jahatef
wants to merge
30
commits into
main
Choose a base branch
from
rwkv-tp
base: main
Could not load branches
Branch not found: {{ refName }}
Loading
Could not load tags
Nothing to show
Loading
Are you sure you want to change the base?
Some commits from the old base branch may be removed from the timeline,
and old review comments may become outdated.
Open
Changes from 1 commit
Commits
Show all changes
30 commits
Select commit
Hold shift + click to select a range
4c7cb11
inital tp commits
jahatef 46904d5
setup
jahatef e2933ef
configs
jahatef d1112ab
merge
jahatef 43d641d
time mixing tp
jahatef de02f37
time-mixing
jahatef dd441b6
time mixing debugging
jahatef a418670
reset time_faaaa
jahatef 540d856
Add additional asserts and update post training readme (#1300)
AI-WAIFU 12aac35
Fix failling tests (#1301)
AI-WAIFU 97c7915
inital tp commits
jahatef 5f89ed8
merge
jahatef 91cb759
Add ERROR logging prefix and sort the prefixes alphabetically (#1308)
TheBatmanofButler 49b263a
inital tp commits
jahatef 48de682
cleanup
jahatef c6fac96
cleanup
jahatef 5a259c0
Update local_setup.yml
jahatef c2d6c85
add Triton FLA
jahatef bdb3658
change version of rwkv-fla
jahatef ff7f328
fix a GQA issue (#1314) (#1315)
tiandeyu-cs 1350b2c
fix 'intermediate_size' in Llama configuration files after the 'mlp_t…
tiandeyu-cs c4d7a54
Python 3.10 support (#1313)
markNZed ee2f142
Fix documentation for converting SFT/DPO weights back to HF Llama (#1…
jacobthebanana 6e81f0b
fix bug (#1311)
AI-WAIFU df95419
Add support for dropout in sparse attention (#1312)
michaelc-yu d682529
adds pyproject files and tests (#1302)
LouisCastricato 0bc11d6
undo merge error (#1325)
Quentin-Anthony c6db95c
inital tp commits
jahatef daac503
setup
jahatef bf478ce
Merge branch 'main' into rwkv-tp
Quentin-Anthony File filter
Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Add ERROR logging prefix and sort the prefixes alphabetically (#1308)
* Add ERROR logging prefix and sort alphabetically * fix comment
- Loading branch information
commit 91cb7590531fde52b68bc952571b6bc8ca0b5ea8
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I don't think we've properly merged this branch onto upstream
main
, since this is tracking as a change. Please do this.