Remove some trailing white space.

This commit is contained in:
comfyanonymous 2024-12-27 18:02:21 -05:00
parent 9cfd185676
commit d170292594
15 changed files with 37 additions and 38 deletions

View File

@ -42,7 +42,6 @@ def bipartite_soft_matching_random2d(metric: torch.Tensor,
gather = mps_gather_workaround if metric.device.type == "mps" else torch.gather gather = mps_gather_workaround if metric.device.type == "mps" else torch.gather
with torch.no_grad(): with torch.no_grad():
hsy, wsx = h // sy, w // sx hsy, wsx = h // sy, w // sx
# For each sy by sx kernel, randomly assign one token to be dst and the rest src # For each sy by sx kernel, randomly assign one token to be dst and the rest src