chong.zhang
2023-03-27 3fe6147415a65313fdc856c94fb92a5eb65a63da
funasr/layers/label_aggregation.py
@@ -79,4 +79,4 @@
        else:
            olens = None
        return output, olens
        return output.to(input.dtype), olens