-
Notifications
You must be signed in to change notification settings - Fork 659
Generated code snippets do not work in colab #1984
Description
code snippet:
!python -m moshi.server --hf-repo "nvidia/personaplex-7b-v1"
error:
[Info] retrieving checkpoint
[Info] loading mimi
Traceback (most recent call last):
File "", line 198, in _run_module_as_main
File "", line 88, in _run_code
File "/usr/local/lib/python3.12/dist-packages/moshi/server.py", line 287, in
main()
File "/usr/local/lib/python3.12/dist-packages/moshi/server.py", line 227, in main
mimi = checkpoint_info.get_mimi(device=args.device)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib/python3.12/dist-packages/moshi/models/loaders.py", line 286, in get_mimi
num_codebooks = max(self.lm_config["dep_q"], self.lm_config["n_q"] - self.lm_config["dep_q"])
~~~~~~~~~~~~~~^^^^^^^^^
KeyError: 'dep_q'
colab:https://colab.research.google.com/#fileId=https%3A//huggingface.co/nvidia/personaplex-7b-v1.ipynb