Update README.md
Browse filesAdded torchvision for compatibility
README.md
CHANGED
|
@@ -59,6 +59,7 @@ Inference using Huggingface transformers on NVIDIA GPUs. Requirements tested on
|
|
| 59 |
|
| 60 |
```
|
| 61 |
torch==2.6.0
|
|
|
|
| 62 |
transformers==4.46.3
|
| 63 |
tokenizers==0.20.3
|
| 64 |
einops
|
|
|
|
| 59 |
|
| 60 |
```
|
| 61 |
torch==2.6.0
|
| 62 |
+
torchvision==0.21.0
|
| 63 |
transformers==4.46.3
|
| 64 |
tokenizers==0.20.3
|
| 65 |
einops
|