formatting

huggingface · Jan 10, 2024 · 1e1a5bb · 1e1a5bb
1 parent d153183
commit 1e1a5bb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/trl/trainer/kto_trainer.py b/trl/trainer/kto_trainer.py
@@ -230,7 +230,7 @@ def make_inputs_require_grad(module, input, output):
         if args.max_completion_length is not None and self.is_encoder_decoder:
             max_completion_length = args.max_completion_length
 
-        if data_collator is None:      
+        if data_collator is None:
             data_collator = DPODataCollatorWithPadding(
                 pad_token_id=tokenizer.pad_token_id,
                 label_pad_token_id=label_pad_token_id,