Update README.md

Add PyTorch requirement to Installation section
cabralpinto · Aug 28, 2023 · b0752e5 · b0752e5
1 parent 6c1e381
commit b0752e5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -22,6 +22,8 @@ Modular Diffusion officially supports Python 3.10+ and is available on PyPI:
 pip install modular-diffusion
 ```
 
+You also need to install the correct [PyTorch distribution](https://pytorch.org/get-started/locally/) for your system.
+
 > **Note**: Although Modular Diffusion works with later Python versions, we currently recommend using Python 3.10. This is because `torch.compile`, which significantly improves the speed of the models, is not currently available for versions above Python 3.10.
 
 ## Usage