yufan
2023-04-23 6000b99579865032977572896da67230d7b2a767
funasr/layers/label_aggregation.py
@@ -79,4 +79,4 @@
        else:
            olens = None
        return output, olens
        return output.to(input.dtype), olens