Update README.md
Browse files
README.md
CHANGED
@@ -44,6 +44,8 @@ git clone https://huggingface.co/Wi-zz/joy-caption-pre-alpha
|
|
44 |
cd joy-caption-pre-alpha
|
45 |
python -m venv venv
|
46 |
.\venv\Scripts\activate
|
|
|
|
|
47 |
pip install -r requirements.txt
|
48 |
```
|
49 |
|
@@ -54,6 +56,7 @@ git clone https://huggingface.co/Wi-zz/joy-caption-pre-alpha
|
|
54 |
cd joy-caption-pre-alpha
|
55 |
python3 -m venv venv
|
56 |
source venv/bin/activate
|
|
|
57 |
pip3 install -r requirements.txt
|
58 |
```
|
59 |
|
|
|
44 |
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 |
+
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
|
49 |
pip install -r requirements.txt
|
50 |
```
|
51 |
|
|
|
56 |
cd joy-caption-pre-alpha
|
57 |
python3 -m venv venv
|
58 |
source venv/bin/activate
|
59 |
+
pip3 install torch torchvision torchaudio
|
60 |
pip3 install -r requirements.txt
|
61 |
```
|
62 |
|