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
|