TouchZZZ
2025-05-22 eccf74684743593950ad259877db7ce9ffc06e93
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":