Open kahkitng98 opened 3 months ago
make sure you're using the correct torch version & the supported dependencies.. torch===2.1.0
same here - using Colab files from Huggingface
I had the same issue, in my case I fixed it by forcing a compatible torch vision 0.16.0 version (matching torch and torch audio = 2.1.0) - make sure to start with clean colab environment, my pip installation looks like this now: !pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu121
There seems to be an error in loading the model when trying to run the codes on Google Colab whereby it gives the following error as attached.