aky15
2023-04-10 d46a542fae26009eee16204a81903862cb4dba73
funasr/layers/label_aggregation.py
@@ -79,4 +79,4 @@
        else:
            olens = None
        return output, olens
        return output.to(input.dtype), olens