Обучал я модель с помощью этого ноутбука https://colab.research.google.com/drive/11TSOWOpj3Xe7GDT3bzO1X9ayRYCwhjFB?usp=sharing
Чтобы насинтезировать вавки, можно использовать infer.ipynb из репозитория, вавки нужно положить в infer_path из конфига, они сохранятся в папку "infer_res". Веса скачаются автоматически.