Error trying to run inference using official Gradio demo code and weights from this repo

#3
by mingyi456 - opened

Hi, I am trying to run the gradio demo code from here: https://huggingface.co/spaces/sesame/csm-1b , after adapting the code to work on a local Windows machine instead of a HuggingFace space. I wanted to try using the model weights over here in order to avoid the access restrictions from the official repo. However, I get this error when trying to load the model weights.

huggingface_hub\hub_mixin.py", line 836, in _load_dataclass
    return datacls(**{k: v for k, v in data.items() if k in datacls.__dataclass_fields__})
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
TypeError: ModelArgs.__init__() missing 4 required positional arguments: 'backbone_flavor', 'decoder_flavor', 'audio_vocab_size', and 'audio_num_codebooks'

This error happens in the line model = Model.from_pretrained(r"unsloth/csm-1b"). Is there something else I need to change in the code to make it work, or is something wrong with my setup?

Sign up or log in to comment