funasr/auto/auto_model.py
@@ -182,7 +182,9 @@ set_all_random_seed(kwargs.get("seed", 0)) device = kwargs.get("device", "cuda") if not torch.cuda.is_available() or kwargs.get("ngpu", 1) == 0: if ((device =="cuda" and not torch.cuda.is_available()) or (device == "xpu" and not torch.xpu.is_available()) or kwargs.get("ngpu", 1) == 0): device = "cpu" kwargs["batch_size"] = 1 kwargs["device"] = device