nichongjia-2007
2023-06-07 c7fc6149b3c5c2de3107c4f1d4983309882d1a1a
funasr/layers/label_aggregation.py
@@ -79,4 +79,4 @@
        else:
            olens = None
        return output, olens
        return output.to(input.dtype), olens