Update requirements.txt
Browse files- requirements.txt +2 -1
requirements.txt
CHANGED
|
@@ -4,4 +4,5 @@ accelerate
|
|
| 4 |
safetensors
|
| 5 |
transformers
|
| 6 |
torchvision
|
| 7 |
-
warmup-scheduler @ git+https://github.com/ildoonet/pytorch-gradual-warmup-lr.git
|
|
|
|
|
|
| 4 |
safetensors
|
| 5 |
transformers
|
| 6 |
torchvision
|
| 7 |
+
warmup-scheduler @ git+https://github.com/ildoonet/pytorch-gradual-warmup-lr.git
|
| 8 |
+
torchtools @ git+https://github.com/pabloppp/pytorch-tools
|