Upload requirements.txt
Browse files- requirements.txt +4 -2
requirements.txt
CHANGED
|
@@ -2,6 +2,7 @@
|
|
| 2 |
torch>=2.0.0
|
| 3 |
transformers>=4.30.0
|
| 4 |
lightning>=2.0.0
|
|
|
|
| 5 |
|
| 6 |
# Web UI
|
| 7 |
gradio>=5.13.1
|
|
@@ -14,5 +15,6 @@ spaces>=0.19.0
|
|
| 14 |
accelerate>=0.20.0
|
| 15 |
bitsandbytes>=0.41.0
|
| 16 |
|
| 17 |
-
# Data visualization
|
| 18 |
-
matplotlib>=3.7.0
|
|
|
|
|
|
| 2 |
torch>=2.0.0
|
| 3 |
transformers>=4.30.0
|
| 4 |
lightning>=2.0.0
|
| 5 |
+
numpy>=1.24.0
|
| 6 |
|
| 7 |
# Web UI
|
| 8 |
gradio>=5.13.1
|
|
|
|
| 15 |
accelerate>=0.20.0
|
| 16 |
bitsandbytes>=0.41.0
|
| 17 |
|
| 18 |
+
# Data visualization and logging
|
| 19 |
+
matplotlib>=3.7.0
|
| 20 |
+
tensorboard>=2.13.0
|