Rustコンパイラで画像生成AI(Stable Diffusion)を試してみた。
意外と簡単でLinux(Ubuntu20.04)で以下のコマンドを入力すると
cargo run --example stable-diffusion --features clap -- --prompt "A rusty robot holding a fire torch."
最近では２万円以下で手に入る廉価版Ryzen 5 5600GでもGPU(CUDA)を使用しないで、CPUだけでも10分程度でAIが画像を生成してくれる。
RustはマルチスレッドでCPU処理ができるので、CPUだけでも結構速い。
画像生成を指示する呪文(prompt)の文章は4種類選んでみた。
"A pretty cat hoding a fire torch."
"A rusty robot holding a fire torch."
"A smart dog hoding a fire torch."
実写でないことはトーチの持ち方でわかるでしょう。
トップ画像の美少女アニメ風の画像を生成する呪文は
girl, hatsune miku,shining white hair,shining blue eyes,gradient shiny hair,sailors,wavy hair,braid bangs,twin tail,floating hair,gingham check,moonlight backlight,luminous iridescent powder light on eye,rainbow light particles,masterpiece, lily,blue flower,aqua color flower,fluttering petals,flower headdress,celtic ribbon,shining butterfly,window,ultra detailed beautiful skin,rainbow splash,floating string
全ての画像はRustでStable Diffusionを実行できるcrateの画像生成AI diffusers-rsを使って生成した。
Rustでの実際のやりかたは次回以降に。