add requiements
Browse files- README.md +6 -0
- requirements.txt +2 -0
README.md
CHANGED
|
@@ -40,6 +40,12 @@ The training setup was `4xA100's 80GB` and took ~6 hours to pretrain and ~13 hou
|
|
| 40 |
---
|
| 41 |
|
| 42 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 43 |
## Training:
|
| 44 |
We will release the training code in some time.
|
| 45 |
|
|
|
|
| 40 |
---
|
| 41 |
|
| 42 |
|
| 43 |
+
## Installing requirements
|
| 44 |
+
```
|
| 45 |
+
pip install -qr https://huggingface.co/Tensoic/Cerule-v0.1/resolve/main/requirements.txt
|
| 46 |
+
```
|
| 47 |
+
|
| 48 |
+
|
| 49 |
## Training:
|
| 50 |
We will release the training code in some time.
|
| 51 |
|
requirements.txt
ADDED
|
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
|
|
|
| 1 |
+
transformers>=4.39.1
|
| 2 |
+
flash_attn
|