nichongjia-2007
2023-05-05 b0b8aba7c079b3af4265e0b25849a847bf9266bf
funasr/modules/nets_utils.py
@@ -595,7 +595,7 @@
    mask = torch.zeros(size, size, device=device, dtype=torch.bool)
    for i in range(size):
        if left_chunk_size <= 0:
        if left_chunk_size < 0:
            start = 0
        else:
            start = max((i // chunk_size - left_chunk_size) * chunk_size, 0)