Skip to content

Commit

Permalink
Merge pull request #42 from kyutai-labs/manukyutai-patch-2
Browse files Browse the repository at this point in the history
Update README.md
  • Loading branch information
adefossez authored Sep 18, 2024
2 parents c127e06 + e507771 commit c1e5d8d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion moshi/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,7 @@ from moshi.models import loaders

mimi_weight = hf_hub_download(loaders.DEFAULT_REPO, loaders.MIMI_NAME)
mimi = loaders.get_mimi(mimi_weight, device='cpu')
mimi.set_num_codebooks(8) # up to 32.
mimi.set_num_codebooks(8) # up to 32 for mimi, but limited to 8 for moshi.

wav = torch.randn(1, 1, 24000 * 10) # should be [B, C=1, T]
with torch.no_grad():
Expand Down

0 comments on commit c1e5d8d

Please sign in to comment.