🎨 Format Python code with psf/black

Adamliu1 · Sep 18, 2024 · 03d9a18 · 03d9a18
1 parent 7b26dbc
commit 03d9a18
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/llm_unlearn_ucl/unlearn_harm.py b/llm_unlearn_ucl/unlearn_harm.py
@@ -281,7 +281,9 @@ def main(args) -> None:
                     attention_mask=batch["attention_mask"].to(device),
                     labels=batch["labels"].to(device),
                 )
-                prob_p = torch.nn.functional.softmax(pretrained_outputs.logits, -1).cpu()
+                prob_p = torch.nn.functional.softmax(
+                    pretrained_outputs.logits, -1
+                ).cpu()
                 pretrained_model_precomputed_normal_outputs_aggregated[-1].append(
                     prob_p
                 )