@

sd_meh/merge_methods.py

@@ -218,7 +218,7 @@ def rotate(a: Tensor, b: Tensor, alpha: float, beta: float, **kwargs):
     a_2d = a.reshape(-1, a.shape[-1]).float()
     b_2d = b.reshape(-1, b.shape[-1]).float()
     svd_driver = "gesvd" if a.is_cuda else None
-    u, _, v_t = torch.linalg.svd(torch.matmul(a_2d.T, b_2d), driver=svd_driver)
+    u, _, v_t = torch.linalg.svd(a_2d.T @ b_2d, driver=svd_driver)
 
     alpha_is_float = alpha != round(alpha)
     if alpha_is_float: