Update requirements.txt: full training pipeline with real datasets
Browse files- requirements.txt +2 -0
requirements.txt
CHANGED
|
@@ -6,3 +6,5 @@ huggingface_hub>=0.22.0
|
|
| 6 |
safetensors
|
| 7 |
imageio>=2.0.0
|
| 8 |
Pillow>=9.0.0
|
|
|
|
|
|
|
|
|
| 6 |
safetensors
|
| 7 |
imageio>=2.0.0
|
| 8 |
Pillow>=9.0.0
|
| 9 |
+
datasets>=2.14.0
|
| 10 |
+
torchvision>=0.16.0
|