🐮

Google colab で創作作家AIをつくる

1 min read

はじめに

zenn に初めて投稿します。
うすいといいます。

現在、以下のサイトで 主に NLP、Transformer を使った創作ノベル関連のツールを作って遊んでいます。

https://novelup.plus/story/942595339

今回、「創作作家AI」という名前でツールを公開したのでご紹介します。

創作作家AI

内容としては、Google Colab を利用しており、事前学習モデルはrinna社が提供してくれている日本語モデルを使用します。
このページ内で、手順などは記載しています。

https://github.com/xbarusui/study_gpt2-japanese_V2

こちらのツールのウリは以下のような点です。

  • Google Colab 内でtextでのファインチューニングをプログラム修正の手間なくできる
  • xsmall モデルを使用しており、学習、推定が速い
  • もし textがなくても、素の状態で文章生成を試せる

事前準備

  • このリポジトリのローカル環境へのクローン
  • rinna の pretrained モデルのインストール
  • Googleアカウント(Google Colaboratory、Googleドライブで使用)

利用方法

  • 手順は各レポジトリの記載をご参照いただければと思います。

Discussion

ログインするとコメントできます