FLUX.1-devを試してみました

Artist

はじめに

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トークンを超える部分は切り捨てられています。

flux.1-dev pixel art

参考

今回使用した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")

コメント