black reformat

mlfoundations · Dec 12, 2023 · bdc42f0 · bdc42f0
1 parent 460c239
commit bdc42f0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/open_lm/utils/transformers/hf_model.py b/open_lm/utils/transformers/hf_model.py
@@ -79,7 +79,7 @@ def forward(
         ```"""
         # decoder outputs consists of (dec_features, layer_state, dec_hidden, dec_attn)
         logits, _ = self.model(input_ids)
-        
+
         if labels is not None:
             # Shift so that tokens < n predict n
             shift_logits = logits[..., :-1, :].contiguous()