はじめに
Google Colab Proに課金して、最近話題の画像生成AI「FLUX.1-dev」を試してみました!
さすがA100は速い!ただ、がしがし減っていくコンピューティングユニットを見るのがつらいです笑。
画像生成
というわけで、Claude 3.5 Sonnetにざっくりとしたイメージを伝えて夏っぽいピクセルアートを作ってもらいました。結構いい感じではないでしょうか?
A sunny beach scene in pixel art style, depicting a refreshing summer day by the sea. The foreground shows golden sand with tiny, pixelated seashells scattered about. A bright blue ocean takes up the middle of the image, with small, white-capped waves rolling towards the shore. In the water, a few colorful pixel art fish can be seen jumping out of the waves. On the left side of the beach, a vibrant green palm tree leans slightly, its pixelated fronds swaying in the breeze. A red and white striped beach umbrella stands on the right, providing shade for a small blue cool box. The sky is a clear, light blue with a few puffy white pixel clouds. A bright yellow sun shines in the top right corner, its rays extending in a blocky pattern. The entire scene is rendered in a crisp, retro game-inspired pixel art style, conveying a sense of summer relaxation and seaside freshness.
※ ただし、文章が長すぎたようで77トークンを超える部分は切り捨てられています。
参考
今回使用したGoogle Colabの生成コードです。
# パッケージのインストール
!pip install git+https://github.com/huggingface/diffusers.git
# HuggingFaceログイン
!huggingface-cli login
# 以下画像生成コード
import torch
from diffusers import FluxPipeline
pipe = FluxPipeline.from_pretrained(
"black-forest-labs/FLUX.1-dev",
torch_dtype=torch.bfloat16
)
pipe.enable_model_cpu_offload()
out = pipe(
prompt="ここに生成したい画像のプロンプトを入力する",
guidance_scale=3.5,
width=1024,
height=1024,
num_inference_steps=50,
).images[0]
out.save("image.png")
コメント