Update README.md
Browse files
README.md
CHANGED
@@ -45,7 +45,7 @@ cd joy-caption-pre-alpha
|
|
45 |
python -m venv venv
|
46 |
.\venv\Scripts\activate
|
47 |
# Change as per https://pytorch.org/get-started/locally/
|
48 |
-
|
49 |
pip install -r requirements.txt
|
50 |
```
|
51 |
|
|
|
45 |
python -m venv venv
|
46 |
.\venv\Scripts\activate
|
47 |
# Change as per https://pytorch.org/get-started/locally/
|
48 |
+
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
|
49 |
pip install -r requirements.txt
|
50 |
```
|
51 |
|