Fix file path for shard_num 1 with mllama converter

[strangiato]

What does this PR do?

mllama conversion with shard_num 1 expects the file to be consolidated.pth but the Llama 3.2 11B uses consolidated.00.pth instead.

Fixes #35049

Before submitting

• This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
• Did you read the contributor guideline,
  Pull Request section?
• Was this discussed/approved via a Github issue or the forum? Please add a link
  to it if that's the case.
• Did you make sure to update the documentation with your changes? Here are the
  documentation guidelines, and
  here are tips on formatting docstrings.
• Did you write any new necessary tests?

Who can review?

Anyone in the community is free to review the PR once the tests have passed. Feel free to tag
members/contributors who may be interested in your PR.

• text models: @ArthurZucker
• vision models: @amyeroberts, @qubvel

[qubvel]

Hey @strangiato, for llama vision models the file is consolidated.pth

see
https://huggingface.co/meta-llama/Llama-3.2-11B-Vision-Instruct/tree/main/original

[strangiato]

It seems there is a discrepancy between the files that is published on HuggingFace for the .pth format, and the files published on meta directly (https://www.llama.com/llama-downloads/)

If you download the files directly form meta they are using consolidated.00.pth not consolidated.pth.

[qubvel]

Ok, then it's better to support both, smth like

path = "consloidated.pth" if os.path.exists("consloidated.pth") else "consloidated.00.pth"

[strangiato]

Updated the code to check if the 00.pth file exists and falls back to the non-numbered one in a similar what to what you suggested.

[zucchini-nlp]

LGTM, thanks !

[HuggingFaceDocBuilderDev]

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.