zhifu gao
2023-04-24 033ef80e4e721cd980bf3e0c4f0c112bab2dd603
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)