Skip to content

Commit fd3730d

Browse files
committed
Updated the README.md with pytorch link
1 parent 25d93c9 commit fd3730d

File tree

1 file changed

+13
-4
lines changed

1 file changed

+13
-4
lines changed

README.md

Lines changed: 13 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,14 @@ AltTextGenerator is a Python script designed to automatically generate alt text
1717
- Hugging Face Transformers (>=4.48.0,<5.0.0) library
1818
- torch (>=2.5.1,<3.0.0) library
1919

20+
## PyTorch Dependencies
21+
22+
To ensure you install the correct versions of torch, torchvision, and torchaudio that are compatible with your system (e.g. with CUDA support or CPU-only), please refer to the official [PyTorch Get Started Locally](https://pytorch.org/get-started/locally/) page.
23+
<br>For example, if you are using pip and wish to install a CUDA-enabled version (replace cu126 with the CUDA version suited to your system), you might run:
24+
```bash
25+
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126
26+
```
27+
2028
## Installation
2129

2230
1. Clone the repository or download the script:
@@ -73,10 +81,11 @@ To ensure the script is working correctly, follow these steps to run the tests:
7381
pytest tests/
7482
```
7583
3. Test details:
76-
- The tests include validation for:
77-
- The BLIP2 model functionality.
78-
- Utility functions for file and image processing.
79-
- End-to-end integration of image processing and renaming.
84+
- The tests include validation for:
85+
- The BLIP2 model functionality.
86+
- Utility functions for file and image processing.
87+
- End-to-end integration of image processing and renaming.<br>
88+
8089
4. Check the output to conform that all tests have passed. Example:
8190
```bash
8291
============================= test session starts ==============================

0 commit comments

Comments
 (0)