Config: warning when saving generation kwargs in the model config (#28514)

2024-01-16 18:31:01 +00:00
parent 7142bdfa90
commit f4f57f9dfa
6 changed files with 107 additions and 32 deletions
--- a/tests/generation/test_framework_agnostic.py
+++ b/tests/generation/test_framework_agnostic.py
@@ -529,7 +529,7 @@ class GenerationIntegrationTestsMixin:

        pixel_values = floats_tensor((2, 3, 30, 30))
        model = model_cls.from_pretrained("hf-internal-testing/tiny-random-VisionEncoderDecoderModel-vit-gpt2")
-        model.config.decoder.eos_token_id = None
+        model.generation_config.eos_token_id = None
        if is_pt:
            pixel_values = pixel_values.to(torch_device)
            model = model.to(torch_device)