add back newline

mosaicml · Dec 7, 2023 · 7f9be51 · 7f9be51
1 parent d05911e
commit 7f9be51
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/scripts/inference/hf_generate.py b/scripts/inference/hf_generate.py
@@ -287,6 +287,7 @@ def _generate(encoded_inp: Dict[str, torch.Tensor]):
                 encoded_inp['input_ids'] !=
                 tokenizer.pad_token_id,  # type: ignore
                 axis=1).numpy(force=True)
+
             # Warmup
             if args.warmup and (not done_warmup):
                 print('Warming up...')