From b0752e5c7903ef7e495ccb80fc27501d82566212 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jo=C3=A3o=20Cabral=20Pinto?=
 <47889626+cabralpinto@users.noreply.github.com>
Date: Mon, 28 Aug 2023 18:47:26 +0100
Subject: [PATCH] Update README.md

Add PyTorch requirement to Installation section
---
 README.md | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/README.md b/README.md
index 6afe631..72276c4 100644
--- 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