请问 PyTorch 需求

#6
by roach231428 - opened

加载模型时遇到错误,请问有PyTorch的需求条件吗?
PyTorch版本是1.13.0

ValueError: Could not load model BELLE-2/Belle-whisper-large-v3-turbo-zh with any of the following classes: (<class 'transformers.models.auto.modeling_auto.AutoModelForCTC'>, <class 'transformers.models.auto.modeling_auto.AutoModelForSpeechSeq2Seq'>, <class 'transformers.models.whisper.modeling_whisper.WhisperForConditionalGeneration'>). 

升级到 PyTorch 2.1.0 就好了,应该是版本问题

Sign up or log in to comment