Skip to content

Commit

Permalink
Merge pull request #503 from simonrouard/patch-1
Browse files Browse the repository at this point in the history
Update MUSICGEN_STYLE.md
  • Loading branch information
JadeCopet authored Nov 11, 2024
2 parents 4ec3e79 + 57e1492 commit f593185
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 f593185

Please sign in to comment.