Skip to content

Commit

Permalink
Fix Distl Whisper error: out of bounds for dimension 0 with size y th…
Browse files Browse the repository at this point in the history
…at occurs … (#2264)

CVS-148233

Problem is in pythorch 2.4.0 on windows on CPU. torch version is
limited.
  • Loading branch information
aleksandr-mokrov authored Aug 5, 2024
1 parent 3673851 commit 74b7b08
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion notebooks/distil-whisper-asr/distil-whisper-asr.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,7 @@
},
"outputs": [],
"source": [
"%pip install -q \"transformers>=4.35\" \"torch>=2.1\" onnx \"git+https://github.com/huggingface/optimum-intel.git\" \"peft==0.6.2\" --extra-index-url https://download.pytorch.org/whl/cpu\n",
"%pip install -q \"transformers>=4.35\" \"torch>=2.1,<2.4.0\" onnx \"git+https://github.com/huggingface/optimum-intel.git\" \"peft==0.6.2\" --extra-index-url https://download.pytorch.org/whl/cpu\n",
"%pip install -q \"openvino>=2023.2.0\" datasets \"gradio>=4.0\" \"librosa\" \"soundfile\"\n",
"%pip install -q \"nncf>=2.6.0\" \"jiwer\""
]
Expand Down

0 comments on commit 74b7b08

Please sign in to comment.