mengzhe.cmz
2023-05-08 f5825f1f4f228e459455e7b739bf3b7e2aac3395
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)