funasr/auto/auto_model.py @@ -233,6 +233,8 @@ # fp16 if kwargs.get("fp16", False): model.to(torch.float16) elif kwargs.get("bf16", False): model.to(torch.bfloat16) return model, kwargs def __call__(self, *args, **cfg):
