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 |
|