Skip to content

Commit

Permalink
add instructions for finetuning
Browse files Browse the repository at this point in the history
  • Loading branch information
jiaqili3 committed Jul 25, 2024
1 parent 5b284e3 commit 71f589a
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions egs/tts/VALLE_V2/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -130,6 +130,10 @@ Run:
sh egs/tts/VALLE_V2/train_ar_libritts.sh --resume
```

### Finetuning based on our AR model
We provide our AR model optimizer, and random_states checkpoints to support finetuning (No need to download these files if you're only inferencing from the pretrained model). First rename the models as "pytorch_model.bin", "optimizer.bin", and "random_states_0.pkl", then you could resume from these checkpoints. [Link to AR optimizer checkpoint](https://huggingface.co/amphion/valle/blob/main/optimizer_valle_ar_mls_196000.bin) and [Link to random_states.pkl](https://huggingface.co/amphion/valle/blob/main/random_states_0.pkl).


### Run the command to Train NAR model
(Make sure your current directory is at the Amphion root directory).
Run:
Expand Down

0 comments on commit 71f589a

Please sign in to comment.