ChaiML
/

gpt2_base_retry_and_continue_5m_reward_model

Text Classification

text-generation-inference

Model card Files Files and versions

Jellywibble commited on Mar 6, 2023

Commit

26229d9

·

1 Parent(s): fd82b87

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -26,7 +26,7 @@ Its training dataset consists of purely user-generated content [retry_and_contin
 ### Out-of-scope use
 ### How to use
-This reward model can be loaded using the `AutoModelForSequenceClassification` functionality, with a GPT2 tokenizer
 ```python
 from transformers import AutoTokenizer, AutoModelForSequenceClassification

 ### Out-of-scope use
 ### How to use
+This reward model can be loaded using the `AutoModelForSequenceClassification` functionality, with a GPT2 tokenizer where the `pad_token_id` is set to the EOS token id, padding sides need to be set according to the configurations used during model training.
 ```python
 from transformers import AutoTokenizer, AutoModelForSequenceClassification