Update README.md
Browse files
README.md
CHANGED
|
@@ -51,7 +51,7 @@ gptj-chai:
|
|
| 51 |
deepspeed trainer_sft.py --local_rank=0 --configs defaults gptj-chai soda-only --cache_dir data_cache --deepspeed
|
| 52 |
```
|
| 53 |
|
| 54 |
-
# Demo
|
| 55 |
```python
|
| 56 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 57 |
|
|
|
|
| 51 |
deepspeed trainer_sft.py --local_rank=0 --configs defaults gptj-chai soda-only --cache_dir data_cache --deepspeed
|
| 52 |
```
|
| 53 |
|
| 54 |
+
# Interactive Demo Code:
|
| 55 |
```python
|
| 56 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 57 |
|