AttributeError: module 'audiocraft' has no attribute '__version__' when running on Google Colab

[kahkitng98]

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.

[Erichy-dev]

make sure you're using the correct torch version & the supported dependencies.. torch===2.1.0

https://pytorch.org/get-started/previous-versions/

[LordMidi]

same here - using Colab files from Huggingface

https://huggingface.co/facebook/musicgen-stereo-medium

[gitbarlew]

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