diff --git a/examples/summarize_rlhf/trlx_gptj_text_summarization.py b/examples/summarize_rlhf/trlx_gptj_text_summarization.py index 9103ed87d..3c2dbe6f2 100755 --- a/examples/summarize_rlhf/trlx_gptj_text_summarization.py +++ b/examples/summarize_rlhf/trlx_gptj_text_summarization.py @@ -97,7 +97,7 @@ def reward_fn(samples: List[str], **kwargs): ) config = TRLConfig.load_yaml(config_path) - tokenizer = AutoTokenizer.from_pretrained(config.model.tokenizer_path) + tokenizer = AutoTokenizer.from_pretrained(config.tokenizer.tokenizer_path) tokenizer.pad_token = tokenizer.eos_token tokenizer.padding_side = "left" max_length_input = (