Fix Swin v2 tuple type hint

pull/1239/head
Ross Wightman 3 years ago
parent bb85b09d2a
commit 15cc9eae3e

@ -117,7 +117,7 @@ def window_partition(x, window_size: Tuple[int, int]):
@register_notrace_function # reason: int argument is a Proxy
def window_reverse(windows, window_size: tuple[int, int], img_size: tuple[int, int]):
def window_reverse(windows, window_size: Tuple[int, int], img_size: Tuple[int, int]):
"""
Args:
windows: (num_windows * B, window_size[0], window_size[1], C)

Loading…
Cancel
Save