From 57e1492e27b6037c2e9be97afa9ffbe1ee2508f8 Mon Sep 17 00:00:00 2001 From: Simon Rouard <74627808+simonrouard@users.noreply.github.com> Date: Mon, 11 Nov 2024 02:03:09 -0800 Subject: [PATCH] Update MUSICGEN_STYLE.md --- docs/MUSICGEN_STYLE.md | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/docs/MUSICGEN_STYLE.md b/docs/MUSICGEN_STYLE.md index 34955b11..9962b2ea 100644 --- a/docs/MUSICGEN_STYLE.md +++ b/docs/MUSICGEN_STYLE.md @@ -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/