cannot import name 'HF_MODEL_CHECKPOINTS_MAP' from 'audiocraft.models.loaders'

[boudini]

Hello,
Thank you for providing this cog

First update, in order for the image to build, you should do these in your run instead in order to re-download the keyrings from nvidia and installing them

commands run after the environment is setup

run:
- "apt-key del 7fa2af80"
- "apt-get update && apt-get install -y --no-install-recommends wget"
- "wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.0-1_all.deb"
- "dpkg -i cuda-keyring_1.0-1_all.deb"
- "pip install pip install -U git+https://git@github.com/facebookresearch/audiocraft"
- "apt-get update && apt-get install -y ffmpeg"

Second, After the image builds successfully, we get this error:
Validating model schema...

Traceback (most recent call last):
File "/root/.pyenv/versions/3.9.18/lib/python3.9/runpy.py", line 197, in _run_module_as_main
return _run_code(code, main_globals, None,
File "/root/.pyenv/versions/3.9.18/lib/python3.9/runpy.py", line 87, in _run_code
exec(code, run_globals)
File "/root/.pyenv/versions/3.9.18/lib/python3.9/site-packages/cog/command/openapi_schema.py", line 18, in
app = create_app(config, shutdown_event=None)
File "/root/.pyenv/versions/3.9.18/lib/python3.9/site-packages/cog/server/http.py", line 72, in create_app
predictor = load_predictor_from_ref(predictor_ref)
File "/root/.pyenv/versions/3.9.18/lib/python3.9/site-packages/cog/predictor.py", line 172, in load_predictor_from_ref
spec.loader.exec_module(module)
File "", line 850, in exec_module
File "", line 228, in _call_with_frames_removed
File "predict.py", line 30, in
from audiocraft.models.loaders import (
ImportError: cannot import name 'HF_MODEL_CHECKPOINTS_MAP' from 'audiocraft.models.loaders' (/root/.pyenv/versions/3.9.18/lib/python3.9/site-packages/audiocraft/models/loaders.py)