fix whitespace for patch check

This commit is contained in:
Wing Lian
2024-12-06 16:43:44 -05:00
parent 2e9e423dfd
commit 31723ac523

View File

@@ -49,6 +49,7 @@ ORIGINAL_LLAMA_FCLM_CODE = """
cache_position=cache_position,
**kwargs,
)
hidden_states = outputs[0]
# Only compute necessary logits, and do not upcast them to float if we are not computing the loss
logits = self.lm_head(hidden_states[:, -num_logits_to_keep:, :])