diff --git a/funasr/models/sense_voice/model_small.py b/funasr/models/sense_voice/model_small.py index 5d187837f..af7c603e9 100644 --- a/funasr/models/sense_voice/model_small.py +++ b/funasr/models/sense_voice/model_small.py @@ -1946,7 +1946,7 @@ class SenseVoiceL(nn.Module): speech = speech.to(device=kwargs["device"])[0, :, :] speech_lengths = speech_lengths.to(device=kwargs["device"]) - DecodingOptions = kwargs.get("DecodingOptions", {}) + DecodingOptions = kwargs.get("DecodingOptions", {"fp16": kwargs.get("fp16", True)}) task = DecodingOptions.get("task", "ASR") if isinstance(task, str): task = [task]