🐮
Google colab で創作作家AIをつくる
はじめに
zenn に初めて投稿します。
うすいといいます。
現在、以下のサイトで 主に NLP、Transformer を使った創作ノベル関連のツールを作って遊んでいます。
今回、「創作作家AI」という名前でツールを公開したのでご紹介します。
創作作家AI
内容としては、Google Colab を利用しており、事前学習モデルはrinna社が提供してくれている日本語モデルを使用します。
このページ内で、手順などは記載しています。
こちらのツールのウリは以下のような点です。
- Google Colab 内でtextでのファインチューニングをプログラム修正の手間なくできる
- xsmall モデルを使用しており、学習、推定が速い
- もし textがなくても、素の状態で文章生成を試せる
事前準備
- このリポジトリのローカル環境へのクローン
- rinna の pretrained モデルのインストール
- Googleアカウント(Google Colaboratory、Googleドライブで使用)
利用方法
- 手順は各レポジトリの記載をご参照いただければと思います。
Discussion