Skip to content

Commit ad78ed8

Browse files
authored
add blurb for AMD's fork (#1997)
as titled, with some other minor updates
1 parent 176498c commit ad78ed8

File tree

2 files changed

+4
-5
lines changed

2 files changed

+4
-5
lines changed

README.md

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -20,12 +20,11 @@
2020

2121

2222
## Latest News
23-
- [2025/10] SkyPilot now supports TorchTitan! See the tutorial [here](https://docs.skypilot.co/en/latest/examples/training/torchtitan.html).
23+
- [2025/11] AMD released an [optimized fork](https://github.com/AMD-AGI/torchtitan-amd/tree/main) of `torchtitan` for AMD GPUs.
24+
- [2025/10] We released `torchtitan` [v0.2.0](https://github.com/pytorch/torchtitan/releases).
25+
- [2025/10] SkyPilot now supports `torchtitan`! See the tutorial [here](https://docs.skypilot.co/en/latest/examples/training/torchtitan.html).
2426
- [2025/07] We published [instructions](/torchtitan/models/README.md) on how to add a model to `torchtitan`.
25-
- [2025/07] We released `torchtitan` [v0.1.0](https://github.com/pytorch/torchtitan/releases), and also set up nightly builds.
2627
- [2025/04] Our paper was accepted by [ICLR 2025](https://iclr.cc/virtual/2025/poster/29620).
27-
- [2025/04] Training the diffusion model [FLUX](torchtitan/models/flux/) with FSDP/HSDP is available as an experiment.
28-
- [2025/04] The frontend implementation of [SimpleFSDP](torchtitan/experiments/simple_fsdp/), a compiler-based FSDP framework, is available as an experiment.
2928
- [2024/12] GPU MODE [lecture](https://www.youtube.com/watch?v=VYWRjcUqW6w) on torchtitan.
3029
- [2024/07] [Presentation](https://pytorch2024.sched.com/event/1fHn3) at PyTorch Conference 2024.
3130

torchtitan/models/flux/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ This step will download the autoencoder model from HuggingFace and save it to th
2525

2626
Run the following command to train the model on a single GPU:
2727
```bash
28-
./torchtitan/experiments/flux/run_train.sh
28+
./torchtitan/models/flux/run_train.sh
2929

3030
```
3131

0 commit comments

Comments
 (0)