AI Rap Lyric Generator

AI ラップ歌詞ジェネレーター

アーティスト名と曲名を使用してラップの歌詞を生成する
JAX を使用した Google Cloud TPU-VM を使用して、複数のアーティストの 10,000 曲以上の曲で GPT2 を微調整

モデルは、ハグフェイス スペースのデモとして展開されました。デモをチェックアウト - ここをクリックしてください

トレーニングに使用されるデータセットはここにあります
この Google Colab ノートブックを使用して予測を行うこともできます。ここをクリックしてください
モデル ファイルを含むコード全体については、ここで Huggingface リポジトリをチェックアウトしてください。

リポジトリ内のファイル:-

1. run_clm_flax.py にはトレーニング スクリプトが含まれています
   
2. run.sh を使用してトレーニング スクリプトを実行でき、そこでハイパーパラメータを直接指定できます。
   
3. genius.py には、genius.com から曲をダウンロードするためのファイルが含まれています
   

デモの動作例