adapt style to predefined style layout

2019-12-25 23:32:44 +01:00
parent deff792bb6
commit fc84bd5254
5 changed files with 8 additions and 8 deletions
--- a/src/transformers/modeling_ctrl.py
+++ b/src/transformers/modeling_ctrl.py
@@ -492,7 +492,7 @@ class CTRLLMHeadModel(CTRLPreTrainedModel):

    def prepare_inputs_for_generation(self, input_ids, **kwargs):
        # only last token for inputs_ids if past is defined in kwargs
-        if 'past' in kwargs and kwargs['past']:
+        if "past" in kwargs and kwargs["past"]:
            input_ids = input_ids[:, -1].unsqueeze(-1)

        inputs = {"input_ids": input_ids}