s2s / arguments_classes /paraformer_stt_arguments.py
andito's picture
andito HF staff
Upload folder using huggingface_hub
c72e80d verified
raw
history blame
546 Bytes
from dataclasses import dataclass, field
@dataclass
class ParaformerSTTHandlerArguments:
paraformer_stt_model_name: str = field(
default="paraformer-zh",
metadata={
"help": "The pretrained model to use. Default is 'paraformer-zh'. Can be choose from https://github.com/modelscope/FunASR"
},
)
paraformer_stt_device: str = field(
default="cuda",
metadata={
"help": "The device type on which the model will run. Default is 'cuda' for GPU acceleration."
},
)