Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
divya-kumari32 authored Dec 18, 2024
1 parent c2943bb commit faccab3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion training/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ We trained our Bamba model with FSDP using our training repo [here](https://gith
Note that this training effort was started before FSDP2 and also long before we contributed
`Mamba2-Hybrid` to HF, so we were doing FSDP1 training with [official Mamba implementation](https://github.com/state-spaces/mamba).
For users trying to reproduce the training you now have much more options with our newly
contributed [HF-version of Mamba2-Hybrid]() (TODO: add link once live).
contributed [HF-version of Mamba2-Hybrid]() (will update this link soon!).

Here are the setup details and command on how you can train the model:

Expand Down

0 comments on commit faccab3

Please sign in to comment.