haoneng.lhn
2023-04-28 dcc4053d3e25ff3962e9a8236a26ef69ae4c4827
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)