はじめに
今回、以前から興味のあった画像生成Webアプリを作ってみました。
どのようなものにするか
以下はぼんやりと思い描いた設計イメージ。
・入力フォームにプロンプトとネガティブプロンプトを入力する。
・生成ボタンを押すとその下に生成した画像が表示される。
・huggingface のモデルIDを画面で入力して切り替えられるようにする。
・GPUを利用して生成できるようにする。←最重要ポイント
Webアプリのベース作成
AIに要件を伝えてWebアプリのベースを作成しました。
すると以下のような画面が作成できました。
ひとまずUIは自分の思い描いたものに！
肝心なGPUの有効化がうまくいっていなかった件
いざ画像を生成しようとしてみたところやたら時間がかかります...
モニタリングソフトを見てもGPU使用率が全く変わっておらず...
CUDAツールキットの導入
原因を調べたところNVIDIAが配布しているCUDAツールキットを導入する必要があったみたいです。
CUDAツールキットを導入してみたところ無事にGPUを利用した画像生成が有効になり速度が大幅に速くなりました！
以下は実際に作成してみた画面で大体７秒ほどで生成できました
まとめ
ということで無事目的は達成したのですがあくまで簡易的なものになるのでここからカスタマイズして機能追加などしていきたいと思います！