adarshxs commited on
Commit
85e501d
1 Parent(s): 0ed425a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -4
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 and Inference:
44
  We will release the training code in some time.
45
 
46
- ### Inference(Is faulty with the below code. Working on it pls gib time):
47
- **Please note that running the inference code at this stage may result in errors**. The proper code for training and inference shall be released soon!
48
- Before running the snippet, you need to install the following dependencies:
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