Convert list to tuple.

2025-12-16 08:27:49 +01:00 · 2023-04-26 11:45:03 +08:00
parent 284a2ea114
commit 91f020ee35
2 changed files with 4 additions and 2 deletions
--- a/inpainter/model/modules/tfocal_transformer_hq.py
+++ b/inpainter/model/modules/tfocal_transformer_hq.py
@@ -128,8 +128,10 @@ def window_partition(x, window_size):
        windows: (B*num_windows, T*window_size*window_size, C)
    """
    B, T, H, W, C = x.shape
+
    x = x.view(B, T, H // window_size[0], window_size[0], W // window_size[1],
               window_size[1], C)
+
    windows = x.permute(0, 2, 4, 1, 3, 5, 6).contiguous().view(
        -1, T * window_size[0] * window_size[1], C)
    return windows