GERADOR DE LÍRICA AI RAP
Gerando letras de rap usando o nome do artista e o título da música
GPT2 ajustado em mais de 10.000 músicas de vários artistas usando Google Cloud TPU-VM usando JAX
O modelo foi implantado como uma demonstração em espaços huggingface. Confira a demonstração - clique aqui
O conjunto de dados usado para treinamento pode ser encontrado aqui
Você pode até usar este notebook Google Colab para fazer previsões, clique aqui
Para todo o código com arquivo de modelo também disponível, confira o repositório huggingface aqui
Arquivos no repositório: -
- run_clm_flax.py contém o script de treinamento
- run.sh pode ser usado para executar o script de treinamento e os hiperparâmetros podem ser especificados diretamente lá
- genial.py contém o arquivo para baixar músicas de genial.com
Um exemplo de como funciona a demonstração
