[개발이야기#04] 파이썬/인공지능- 나의 노트북에서 AI를 이용하여 이미지 생성하기(feat. anpigon님)

글 작성일: 2024년 9월 12일 (목)

---

안녕하세요. 요거트 @yoghurty 입니다!

오늘 개발이야기는 약간 옆길로 샜습니다. 😆

나의 로컬 PC에서 인공지능(AI)을 이용한 고품질 이미지를 생성할 수 있다는 @anpigon 님의 포스팅을 보고,
재미있을 것 같아 바로 따라해봤습니다!

👉 M2 맥북 Pro에서 Flux Schnell 로컬 모델로 고품질 이미지 생성하기

따라하기

---

1. Flux Schnell 실습을 위한 conda 환경을 설정하고 활성화 시킵니다.

conda create -n flux python=3.11
conda activate flux

2. AI 그림을 그리는데 필요한 패키지 들을 설치합니다.

pip install torch==2.3.1
pip install git+https://github.com/huggingface/diffusers.git@v0.30.1
pip install transformers==4.43.3 sentencepiece==0.2.0 accelerate==0.33.0 protobuf==5.27.3

3. python 코드를 작성합니다.

• M2 맥북용 코드입니다^^

import torch
from diffusers import FluxPipeline
import diffusers

_flux_rope = diffusers.models.transformers.transformer_flux.rope

def new_flux_rope(pos: torch.Tensor, dim: int, theta: int) -> torch.Tensor:
    assert dim % 2 == 0, "The dimension must be even."
    if pos.device.type == "mps":
        return _flux_rope(pos.to("cpu"), dim, theta).to(device=pos.device)
    else:
        return _flux_rope(pos, dim, theta)

diffusers.models.transformers.transformer_flux.rope = new_flux_rope
pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", revision="refs/pr/1", torch_dtype=torch.bfloat16).to("mps")
prompt = "A cat holding a sign that says hello world"
out = pipe(prompt=prompt,  guidance_scale=0.0, height=1024,  width=1024, num_inference_steps=4,  max_sequence_length=256,).images[0]
out.save("flux_image.png")

Hello World!

---

prompt = "A cat holding a sign that says hello world"

prompt 변경하여 다른 그림 그려보기

---

Flux Schnell 모델을 로딩하여 그림을 그려주는 python 코드의 prompt 를 변경해 보았습니다.

prompt = "Santa Claus holding a sign that says Merry Christmas, snowy background, Christmas atmosphere"

ChatGPT (Dall-e)로 그린 그림

ChatGPT로 그리면 어떤 그림이 나올까 갑자기 궁금해졌습니다. ^^
같은 prompt를 이용해서 ChatGPT가 그린 그림입니다.

마무리

---

첫 그림을 그릴때는 시간이 꽤 많이 걸렸는데,
두 번째 그림 부터는 왠지 그림 그리는 속도가 많이 단축된 느낌이 들었습니다.

다양한 prompt를 사용해 보지 않았고,
그림을 많이 그려본 것은 아니지만,
ChatGPT와 견주어도 손색없을 만큼 정말 고품질의 이미지 같은 생각이 드네요! ^^

AI그림 그려주는 다른 구독 서비스 받지 않고,
나의 로컬 PC 에서 내가 직접 AI를 이용해 그림을 그려냈다니~ 신기했습니다! ^^

좋은 기회를 주신 @anpigon 님께 감사드립니다! ^^

이상 요거트 였습니다!

고맙습니다. ^^