Better filtering of the model outputs in Trainer (#8633)

* Better filtering of the model outputs in Trainer * Fix examples tests * Add test for Lysandre
2020-11-19 10:43:15 -05:00
parent f2e07e7272
commit 4208f496ee
16 changed files with 119 additions and 15 deletions
--- a/src/transformers/models/ctrl/configuration_ctrl.py
+++ b/src/transformers/models/ctrl/configuration_ctrl.py
@@ -77,6 +77,7 @@ class CTRLConfig(PretrainedConfig):
    """

    model_type = "ctrl"
+    keys_to_ignore_at_inference = ["past_key_values"]

    def __init__(
        self,