请问 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 就好了,应该是版本问题