Update README.md (CarperAI#180)

lzy37ld · Jan 10, 2023 · 24b041a · 24b041a
1 parent f71401e
commit 24b041a
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -34,6 +34,11 @@ trainer = trlx.train('EleutherAI/gpt-j-6B', dataset=[('dolphins', 'geese'), (1.0
 trainer.generate(**tokenizer('Q: Who rules the world? A:', return_tensors='pt'), do_sample=True)
 ```
 
+#### Save the resulting model to a Hugging Face pretrained language model. (Ready to upload to the Hub!)
+```python
+trainer.save('/path/to/output/folder/')
+```
+
 #### Use 🤗 Accelerate to launch distributed training
 
 ```bash