王梦迪
2025-05-26 9038340be707aa7cbf62fcbf33ab615bb266abdb
funasr/frontends/fused.py
@@ -78,6 +78,10 @@
        self.factors = [frontend.hop_length // self.gcd for frontend in self.frontends]
        if torch.cuda.is_available():
            dev = "cuda"
        elif torch.xpu.is_available():
            dev = "xpu"
        elif torch.backends.mps.is_available():
            dev = "mps"
        else:
            dev = "cpu"
        if self.align_method == "linear_projection":