Update requirements.txt
Browse files- requirements.txt +1 -0
requirements.txt
CHANGED
|
@@ -2,6 +2,7 @@ Pillow
|
|
| 2 |
numpy
|
| 3 |
torch
|
| 4 |
onnxruntime
|
|
|
|
| 5 |
safetensors
|
| 6 |
peft>=0.13.0
|
| 7 |
transformers>=4.36
|
|
|
|
| 2 |
numpy
|
| 3 |
torch
|
| 4 |
onnxruntime
|
| 5 |
+
onnxruntime-gpu
|
| 6 |
safetensors
|
| 7 |
peft>=0.13.0
|
| 8 |
transformers>=4.36
|