Skip to content

Commit

Permalink
Fix typo in supported models table (#864)
Browse files Browse the repository at this point in the history
  • Loading branch information
ashors1 authored May 30, 2024
1 parent 00f755e commit 0621855
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -263,7 +263,7 @@ We currently support the following frameworks and models. More details about eac
| Framework | Supported Models | Use-cases | Container |
| :--- | :---: | :---: | :---: |
| [Paxml](rosetta/rosetta/projects/pax) | GPT, LLaMA, MoE | pretraining, fine-tuning, LoRA | `nvcr.io/nvidia/jax:24.04-paxml-py3` |
| [T5X](rosetta/rosetta/projects/t5x) | T5 | pre-training, fine-tuning | `nvcr.io/nvidia/jax:23.10-paxml-py3` |
| [T5X](rosetta/rosetta/projects/t5x) | T5 | pre-training, fine-tuning | `nvcr.io/nvidia/jax:23.10-t5x-py3` |
| [T5X](rosetta/rosetta/projects/vit) | ViT | pre-training, fine-tuning | `ghcr.io/nvidia/t5x:vit-2023-07-21` |
| [T5X](rosetta/rosetta/projects/imagen) | Imagen | pre-training | `ghcr.io/nvidia/t5x:imagen-2023-10-02` |
| levanter | GPT, LLaMA, MPT, Backpacks | pretraining, fine-tuning | `ghcr.io/nvidia/jax:levanter-2024-04-16` |
Expand Down

0 comments on commit 0621855

Please sign in to comment.