Skip to content

Commit

Permalink
Update MUSICGEN_STYLE.md
Browse files Browse the repository at this point in the history
  • Loading branch information
simonrouard authored Nov 11, 2024
1 parent 4ec3e79 commit 57e1492
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions docs/MUSICGEN_STYLE.md
Original file line number Diff line number Diff line change
Expand Up @@ -194,3 +194,7 @@ dora run solver=musicgen/musicgen_style_32khz model/lm/model_scale=medium contin
to change some parts, like the conditioning or some other parts of the model, you are responsible for manually crafting a checkpoint file from which we can safely run `load_state_dict`.
If you decide to do so, make sure your checkpoint is saved with `torch.save` and contains a dict
`{'best_state': {'model': model_state_dict_here}}`. Directly give the path to `continue_from` without a `//pretrained/` prefix.


[arxiv]: https://arxiv.org/abs/2407.12563
[musicgen_samples]: https://musicgenstyle.github.io/

0 comments on commit 57e1492

Please sign in to comment.