format w/ black

2025-05-06 21:38:25 -07:00
parent 20fc9a8b63
commit 8f24ea16ab
5 changed files with 14 additions and 6 deletions
@@ -25,7 +25,9 @@ def project(
    v1 = torch.nn.functional.normalize(v1, dim=dims)
    v0_parallel = (v0 * v1).sum(dim=dims, keepdim=True) * v1
    v0_orthogonal = v0 - v0_parallel
-    return v0_parallel.to(dtype).to(device_type), v0_orthogonal.to(dtype).to(device_type)
+    return v0_parallel.to(dtype).to(device_type), v0_orthogonal.to(dtype).to(
+        device_type
+    )


 def apg_forward(