Update README.md
Browse files
README.md
CHANGED
@@ -40,17 +40,18 @@ 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 |
|
46 |
-
### Inference
|
47 |
-
|
48 |
-
|
49 |
|
50 |
```shell
|
51 |
pip install torch transformers accelerate pillow
|
52 |
```
|
53 |
|
|
|
54 |
```python
|
55 |
import torch
|
56 |
import transformers
|
|
|
40 |
---
|
41 |
|
42 |
|
43 |
+
## Training:
|
44 |
We will release the training code in some time.
|
45 |
|
46 |
+
### Inference:
|
47 |
+
Clone the following repo and following instructions for a CLI based inference.
|
48 |
+
https://github.com/Tensoic-AI/Cerule
|
49 |
|
50 |
```shell
|
51 |
pip install torch transformers accelerate pillow
|
52 |
```
|
53 |
|
54 |
+
*The below code might break! Please use the CLI based inference in the meantime*
|
55 |
```python
|
56 |
import torch
|
57 |
import transformers
|