shixian.shi
2023-03-15 f59a72d24e917fb2e9560fa646ae80285dba6c95
funasr/layers/label_aggregation.py
@@ -79,4 +79,4 @@
        else:
            olens = None
        return output, olens
        return output.to(input.dtype), olens