Update README.md

README.md

@@ -44,6 +44,8 @@ git clone https://huggingface.co/Wi-zz/joy-caption-pre-alpha
 cd joy-caption-pre-alpha
 python -m venv venv
 .\venv\Scripts\activate
+# Change as per https://pytorch.org/get-started/locally/
+pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
 pip install -r requirements.txt
 ```
 
@@ -54,6 +56,7 @@ git clone https://huggingface.co/Wi-zz/joy-caption-pre-alpha
 cd joy-caption-pre-alpha
 python3 -m venv venv
 source venv/bin/activate
+pip3 install torch torchvision torchaudio
 pip3 install -r requirements.txt
 ```