Merge branch 'main' of https://huggingface.co/KangarooGroup/kangaroo
Browse files
README.md
CHANGED
|
@@ -25,8 +25,10 @@ We introduce <strong>Kangaroo</strong>, a powerful Multimodal Large Language Mod
|
|
| 25 |
|
| 26 |
### Requirements
|
| 27 |
- python == 3.9
|
| 28 |
-
- torch == 2.1.
|
| 29 |
-
- CUDA == 12.1
|
|
|
|
|
|
|
| 30 |
- transformers == 4.41.0
|
| 31 |
- xformers == 0.0.23
|
| 32 |
|
|
|
|
| 25 |
|
| 26 |
### Requirements
|
| 27 |
- python == 3.9
|
| 28 |
+
- torch == 2.1.1, torchvision == 0.16.1
|
| 29 |
+
- CUDA == 12.1
|
| 30 |
+
- timm == 0.9.16
|
| 31 |
+
- flash-attn == 2.6.1
|
| 32 |
- transformers == 4.41.0
|
| 33 |
- xformers == 0.0.23
|
| 34 |
|