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":