Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

model : supporting Bark small model #149

Merged
merged 15 commits into from
Apr 20, 2024
Merged

model : supporting Bark small model #149

merged 15 commits into from
Apr 20, 2024

Conversation

PABannier
Copy link
Owner

@PABannier PABannier commented Apr 16, 2024

This PR adds examples and instructions to support loading the Bark small model.
The bulk of the work consists of writing a correct convert.py script to parse the model checkpoints properly.

Warning: this PR will introduce a breaking change
Bark.cpp will now read Bark weights from a single file which contains all of the weights (the 3 GPT weights + the neural codec weights). We will no longer support 3 different files + the encodec weight. This should give a better user experience, and simplify the starting instructions.

  • Update the convert.py file to support both Bark small or Bark large (no need for an example)
  • Update the instructions in the README

Close #151

@PABannier PABannier marked this pull request as ready for review April 20, 2024 13:31
@PABannier PABannier merged commit 1786f23 into main Apr 20, 2024
2 checks passed
@PABannier PABannier deleted the bark_small branch April 20, 2024 13:35
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

api: merge ggml Bark weights and Encodec weights into a single file
1 participant