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 |
|