Compare commits

...

1 Commits

Author SHA1 Message Date
Sayak Paul
ed53ecadf5 Update distributed_inference.md to correct syntax 2025-12-11 13:46:43 +05:30

View File

@@ -297,9 +297,9 @@ if __name__ == "__main__":
The script above needs to be run with a distributed launcher, such as [torchrun](https://docs.pytorch.org/docs/stable/elastic/run.html), that is compatible with PyTorch. `--nproc-per-node` is set to the number of GPUs available.
/```shell
`torchrun --nproc-per-node 2 above_script.py`.
/```
```shell
torchrun --nproc-per-node 2 above_script.py
```
### Ulysses Attention
@@ -332,4 +332,4 @@ transformer = AutoModel.from_pretrained(
pipeline = DiffusionPipeline.from_pretrained(
CKPT_ID, transformer=transformer, torch_dtype=torch.bfloat16,
).to(device)
```
```