update readme and requirements
Browse files- README.md +10 -0
- requirements.txt +2 -0
README.md
CHANGED
@@ -39,6 +39,16 @@ The training setup was `4xA100's 80GB` and took ~6 hours to pretrain and ~13 hou
|
|
39 |
| ![extreme_ironing](examples/extreme_ironing.jpg) | **What's funny about this image?**<br>The image is quite humorous as it depicts a man ironing clothes on the back of a yellow taxi cab. This is not a typical sight you'd expect to see in everyday life. |
|
40 |
---
|
41 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
42 |
|
43 |
## Training:
|
44 |
We will release the training code in some time.
|
|
|
39 |
| ![extreme_ironing](examples/extreme_ironing.jpg) | **What's funny about this image?**<br>The image is quite humorous as it depicts a man ironing clothes on the back of a yellow taxi cab. This is not a typical sight you'd expect to see in everyday life. |
|
40 |
---
|
41 |
|
42 |
+
## Loading the model
|
43 |
+
|
44 |
+
```
|
45 |
+
pip install -qr https://huggingface.co/Tensoic/Cerule-v0.1/resolve/main/requirements.txt
|
46 |
+
```
|
47 |
+
|
48 |
+
```python
|
49 |
+
from transformers import AutoModelForCausalLM
|
50 |
+
model = AutoModelForCausalLM.from_pretrained("Tensoic/Cerule-v0.1", trust_remote_code=True)
|
51 |
+
```
|
52 |
|
53 |
## Training:
|
54 |
We will release the training code in some time.
|
requirements.txt
ADDED
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
1 |
+
flash_attn
|
2 |
+
transformers>=4.39.1
|