## Installation

```bash
# Install PyTorch (replace cu124 with your local CUDA version)
pip install --upgrade pip
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124

# Install other dependencies
pip install diffusers accelerate fire einops pytorchvideo tqdm imageio matplotlib
```

---

## Quick Start
```bash
# Replace N with the number of available GPUs
torchrun --nproc_per_node=N train.py
```
