RuntimeError and TypeError on Mac

#18
by myyim - opened

I am a mac user, following the code provided on the website using pipeline api and transformers package.
When I run the model using mps, I got "RuntimeError: number of heads in query/key/value should match"
When I run the model using cpu, I got "TypeError: Gemma3nModel.get_placeholder_mask() missing 1 required positional argument: 'audio_features'"
What's the fix?

Google org

Hi @myyim ,

Thanks for reaching out to us, could you please provide me with the reproducible code to assist you better way, I really appreciate your patience in this matter. I have ran the model card code with T4 GPU in Colab. It's running without any issues please find the attached gist file for your reference.

Thanks.

Sign up or log in to comment