Unverified Commit 829e8894 authored by Joao Gante's avatar Joao Gante Committed by GitHub
Browse files

Generate: post-generate config doctest fix (#20804)

* fix doctests

* revert unwanted change
parent 39e620c1
...@@ -2024,7 +2024,7 @@ class GenerationMixin: ...@@ -2024,7 +2024,7 @@ class GenerationMixin:
>>> model = AutoModelForCausalLM.from_pretrained("gpt2") >>> model = AutoModelForCausalLM.from_pretrained("gpt2")
>>> # set pad_token_id to eos_token_id because GPT2 does not have a PAD token >>> # set pad_token_id to eos_token_id because GPT2 does not have a PAD token
>>> model.config.pad_token_id = model.config.eos_token_id >>> model.generation_config.pad_token_id = model.generation_config.eos_token_id
>>> input_prompt = "It might be possible to" >>> input_prompt = "It might be possible to"
>>> input_ids = tokenizer(input_prompt, return_tensors="pt").input_ids >>> input_ids = tokenizer(input_prompt, return_tensors="pt").input_ids
...@@ -2032,7 +2032,7 @@ class GenerationMixin: ...@@ -2032,7 +2032,7 @@ class GenerationMixin:
>>> # instantiate logits processors >>> # instantiate logits processors
>>> logits_processor = LogitsProcessorList( >>> logits_processor = LogitsProcessorList(
... [ ... [
... MinLengthLogitsProcessor(10, eos_token_id=model.config.eos_token_id), ... MinLengthLogitsProcessor(10, eos_token_id=model.generation_config.eos_token_id),
... ] ... ]
... ) ... )
>>> stopping_criteria = StoppingCriteriaList([MaxLengthCriteria(max_length=20)]) >>> stopping_criteria = StoppingCriteriaList([MaxLengthCriteria(max_length=20)])
...@@ -2272,7 +2272,7 @@ class GenerationMixin: ...@@ -2272,7 +2272,7 @@ class GenerationMixin:
>>> # instantiate logits processors >>> # instantiate logits processors
>>> logits_processor = LogitsProcessorList( >>> logits_processor = LogitsProcessorList(
... [ ... [
... MinLengthLogitsProcessor(15, eos_token_id=model.config.eos_token_id), ... MinLengthLogitsProcessor(15, eos_token_id=model.generation_config.eos_token_id),
... ] ... ]
... ) ... )
>>> # instantiate logits processors >>> # instantiate logits processors
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment