19 Commits

Author SHA1 Message Date
Kakaru 39fd90bf44 perf: change 'plt.pcolor' to 'plt.pcolormesh' (#302) 2026-05-24 22:11:36 +08:00
yqzhishen 44ce312264 Full implementation of multi-dictionary support (#238)
* Add multi-dictionary preprocessing and training

* Fix lang_map.json copy

* Add language embed (inject to txt_embed) for acoustic models

* Save language sequence in variance preprocessing

* Display merged phoneme groups properly in distribution plots

* Add multi-dictionary inference

* Save original phoneme texts for duration plots

* Fix duration plots displaying bug

* Explicit `languages` argument passing

* Add language embed (inject to txt_embed) for variance models

* Fix argument passing

* Add log for lang_map.json copy

* Add language embedding scale

* Add language embedding type

* Preprocessing: only apply lang embed on cross-lingual phonemes

* Inference: only apply lang embed on cross-lingual phonemes

* Revert "Add language embedding type"

This reverts commit 655e9ba9611861793297956e79ccbf329313d7f2.

* Revert lang_embed_scale

* Adapt ONNX exporters for multi-language models

* Refactor configuration schemas for datasets

* Add check of existence for merged phonemes

* Fix spk_id assignment

* Fix languages.json filename

* Fix `languages` key in dsconfig.yaml

* Set `use_lang_id` to false if there are no cross-lingual phonemes

* Support defining extra phonemes

* Refactor configs

* Prefer file copies in work_dir when loading dictionaries

* Fix cannot locate dictionary

* Fix unexpected loading error when dictionary changes

* Update toplevel.py (#219)

* Fix unexpected config passing

* Update lynxnet backbone (#228)

* Change the injection method of conditions on lynxnet (#225)

* update configurations for new-lynxnet

* update configurations for new-lynxnet

* update configurations for new-lynxnet

---------

Co-authored-by: KakaruHayate <97896816+KakaruHayate@users.noreply.github.com>

* Improve fastspeech2 encoder using Rotary Position Embedding (RoPE) in multi-head self-attention (#234)

* update multi-head self attention with RoPE

* RoPE onnx (#230)

* fix requirements.txt (#233)

* fix rope for melody encoder

* support swiglu activation for ffn

* update dependencies

---------

Co-authored-by: KakaruHayate <97896816+KakaruHayate@users.noreply.github.com>

* support mini-nsf-hifigan vocoder

* discard negative pad

* fix MHA inference using low torch version

* Fix missing phoneme list sorting

* Fix single-language dictionary parsing language tag

* Add `pitch_controllable` flag to vocoder exporter

(cherry picked from commit a6deb6b5c3)

* support noise injection

* Allow merging global phonemes and language-specific phonemes

* Check for conflicts between short names and global tags

* Finish documentation for multi-dictionary

---------

Co-authored-by: Anjo <87346264+AnAndroNerd@users.noreply.github.com>
Co-authored-by: yxlllc <33565655+yxlllc@users.noreply.github.com>
Co-authored-by: KakaruHayate <97896816+KakaruHayate@users.noreply.github.com>
Co-authored-by: yxlllc <llc1995@sina.com>
2025-03-29 22:19:59 +08:00
yqzhishen 37c0d9533e Restore plot sizes on TensorBoard 2023-11-02 14:44:00 +08:00
Dachun Sun 36ac5f583b Multi-node batched validation and improvement on strategy selection (#148)
* More metadata when binarize

* Picklable AttrDict

* Support other notion of 'sizes'

* Better strategy getter, Ds TB logger, and eval batch sampler.

* Add title to plots and specify figsize

* Unify batch sampler and bug fixes

* Batch and multi-device validation

* Fix imports

* Fix deadlock under multigpu and resume from ckpt

* Remove unnecessary if in base_dataset

* Move build loss to finish init and ft to build model

* Prevent repeated valid item

* Remove optimizer_idx to support lightning 2.1

* Warning message fix

* Fix val error when aux is off

* Rename fields in metadata and add doc

* Adjust respective duration logging for each speaker

* val persisent worker, module list ordering, update doc

---------

Co-authored-by: yqzhishen <yangqian_1015@icloud.com>
2023-10-29 00:37:14 +08:00
yqzhishen 3ae0e7db53 Implement melody encoder and support glide input (#143)
* Add melody encoder and support glide notes

* Support melody encoder inference

* Add note visualization on TensorBoard

* Remove base pitch embedding when using melody encoder

* Fix KeyError `use_glide_embed`

* Fix invalid access to NoneType

* Support melody encoder ONNX export

* Fix KeyError `use_melody_encoder`

* Support note glide ONNX export

* Support custom glide types

* Support glide embedding scale

* Adjust glide type format

* Remove unnecessary interpolation on frame-level MIDI pitch

* Add glide type coverage checks
2023-10-08 23:24:17 +08:00
yqzhishen 9fcfd2efb1 Add phoneme coverage checks to variance binarizer 2023-07-14 22:49:26 +08:00
yqzhishen 96f0f6f5a1 Swap order of gt and pred curve 2023-05-16 18:43:28 +08:00
yqzhishen 88cb146a48 Add energy prediction, re-organize losses 2023-04-28 23:30:17 +08:00
yqzhishen 872f2bcc34 Fix bug 2023-04-17 22:55:42 +08:00
yqzhishen fcce2a14ac Add base pitch to plot 2023-04-17 22:48:52 +08:00
yqzhishen a4fdce7f02 Set grid width to 1 2023-04-17 01:41:23 +08:00
yqzhishen 130b2072af Add grid 2023-04-17 00:37:10 +08:00
yqzhishen b7b927f544 Change color 2023-04-16 20:33:01 +08:00
yqzhishen 9834a0887c Add fs2 pitch predictor and pitch plot 2023-04-16 13:56:28 +08:00
yqzhishen 7d4c31e775 Add alignment dashed lines 2023-04-15 22:45:49 +08:00
yqzhishen 4868b9b396 Add plot_dur 2023-04-15 21:16:16 +08:00
yqzhishen fa55526f89 Tight layout 2023-03-03 20:11:19 +08:00
yqzhishen 02587e07d5 Change plot size 2023-03-03 19:23:04 +08:00
jinglin ce7abdc165 first commit 2021-12-19 11:51:03 +08:00