Can't use custom models - error

[dalmackay]

Traceback (most recent call last):
File "C:\Users\dalli\Desktop\audiocraft plus installers\audiocraft_plus\venv\lib\site-packages\gradio\queueing.py", line 407, in call_prediction
output = await route_utils.call_process_api(
File "C:\Users\dalli\Desktop\audiocraft plus installers\audiocraft_plus\venv\lib\site-packages\gradio\route_utils.py", line 226, in call_process_api
output = await app.get_blocks().process_api(
File "C:\Users\dalli\Desktop\audiocraft plus installers\audiocraft_plus\venv\lib\site-packages\gradio\blocks.py", line 1550, in process_api
result = await self.call_function(
File "C:\Users\dalli\Desktop\audiocraft plus installers\audiocraft_plus\venv\lib\site-packages\gradio\blocks.py", line 1185, in call_function
prediction = await anyio.to_thread.run_sync(
File "C:\Users\dalli\Desktop\audiocraft plus installers\audiocraft_plus\venv\lib\site-packages\anyio\to_thread.py", line 56, in run_sync
return await get_async_backend().run_sync_in_worker_thread(
File "C:\Users\dalli\Desktop\audiocraft plus installers\audiocraft_plus\venv\lib\site-packages\anyio_backends_asyncio.py", line 2144, in run_sync_in_worker_thread
return await future
File "C:\Users\dalli\Desktop\audiocraft plus installers\audiocraft_plus\venv\lib\site-packages\anyio_backends_asyncio.py", line 851, in run
result = context.run(func, *args)
File "C:\Users\dalli\Desktop\audiocraft plus installers\audiocraft_plus\venv\lib\site-packages\gradio\utils.py", line 661, in wrapper
response = f(*args, **kwargs)
File "C:\Users\dalli\Desktop\audiocraft plus installers\audiocraft_plus\app.py", line 914, in predict_full
outs, outs_audio, outs_backup, input_length = _do_predictions(
File "C:\Users\dalli\Desktop\audiocraft plus installers\audiocraft_plus\app.py", line 584, in _do_predictions
outputs = MODEL.generate(texts, progress=progress, return_tokens=USE_DIFFUSION)
File "C:\Users\dalli\Desktop\audiocraft plus installers\audiocraft_plus\audiocraft\models\musicgen.py", line 181, in generate
return self.generate_audio(tokens)
File "C:\Users\dalli\Desktop\audiocraft plus installers\audiocraft_plus\audiocraft\models\musicgen.py", line 403, in generate_audio
gen_audio = self.compression_model.decode(gen_tokens, None)
File "C:\Users\dalli\Desktop\audiocraft plus installers\audiocraft_plus\audiocraft\models\encodec.py", line 356, in decode
res = self.model.decode(codes[None], scales)
File "C:\Users\dalli\Desktop\audiocraft plus installers\audiocraft_plus\venv\lib\site-packages\transformers\models\encodec\modeling_encodec.py", line 738, in decode
audio_values = self._decode_frame(audio_codes[0], audio_scales[0])
File "C:\Users\dalli\Desktop\audiocraft plus installers\audiocraft_plus\venv\lib\site-packages\transformers\models\encodec\modeling_encodec.py", line 702, in _decode_frame
embeddings = self.quantizer.decode(codes)
File "C:\Users\dalli\Desktop\audiocraft plus installers\audiocraft_plus\venv\lib\site-packages\transformers\models\encodec\modeling_encodec.py", line 435, in decode
layer = self.layers[i]
File "C:\Users\dalli\Desktop\audiocraft plus installers\audiocraft_plus\venv\lib\site-packages\torch\nn\modules\container.py", line 293, in getitem
return self._modules[self._get_abs_string_index(idx)]
File "C:\Users\dalli\Desktop\audiocraft plus installers\audiocraft_plus\venv\lib\site-packages\torch\nn\modules\container.py", line 283, in _get_abs_string_index
raise IndexError(f'index {idx} is out of range')
IndexError: index 4 is out of range