Stable Diffusion을 사용하다 보면 자주 보이는 용어 중 하나가 바로 ‘가중치(weights)’입니다. 가중치란 간단히 말해 “특정 요소가 결과물에 얼마나 강력한 영향을 미치는지”를 나타내는 숫자라고 생각하면 쉽습니다.

가중치를 왜 사용하나요?

가중치는 AI가 이미지를 생성할 때 프롬프트 내에서 어떤 키워드나 문구가 더 중요하게 다뤄져야 하는지를 결정합니다. 예를 들어, 프롬프트에 “아름다운 해변”과 “일몰”이라는 두 키워드가 있을 때, 일몰의 색감과 느낌을 더 강조하고 싶다면 “일몰” 키워드에 더 높은 가중치를 부여할 수 있습니다.

• 예시 프롬프트:

아름다운 해변, 일몰:1.5

여기서 ‘일몰’에 붙은 숫자 1.5는 기본 가중치 1보다 1.5배 더 중요하게 반영하라는 뜻입니다.

가중치는 어떻게 표현하나요?

Stable Diffusion에서는 주로 프롬프트 뒤에 콜론(:)과 숫자를 사용하여 가중치를 표시합니다.

• 예시:

고양이:1.2, 귀여운:0.8, 선글라스:1.5

이 프롬프트는 “선글라스”를 가장 강조하고, “귀여운” 요소는 조금 덜 강조하라는 의미가 됩니다.

가중치를 잘 활용하는 팁

가중치를 적절히 활용하면 결과 이미지의 디테일과 분위기를 원하는 대로 정확하게 조정할 수 있습니다. 그러나 너무 과도한 가중치는 오히려 부자연스러운 결과물을 만들 수 있으므로, 0.8~1.5 사이의 값을 권장합니다.

• 가중치 조정 시 참고사항:
  • 높은 가중치 (1.2 이상): 매우 강조하고 싶은 요소
  • 낮은 가중치 (1 이하): 덜 강조하거나 배경 요소로 처리할 요소

Python 코드로 Stable Diffusion에서 가중치를 사용하는 예시

Stable Diffusion을 Python으로 다룰 때 가중치를 설정하는 방법은 다음과 같습니다:

from diffusers import StableDiffusionPipeline
import torch

# 모델 로드
model_id = "CompVis/stable-diffusion-v1-4"
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
pipe = pipe.to("cuda")

# 프롬프트 설정 (가중치 예시)
prompt = "a very beautiful portrait girl:1.2, looking_at_viewer:1.0, blush:0.9, red brown hair:1.3, brown watery eyes:1.1, short smile, shirt, indoors, highly detailed"

# 이미지 생성
image = pipe(prompt).images[0]

# 이미지 저장
image.save("output.png")

위 이미지를 보면, 프롬프트 내에 특정 요소에 가중치를 다르게 부여하여 미세하게 표현된 디테일의 차이를 확인할 수 있습니다. 예를 들어, “red brown hair”의 가중치를 높이면 헤어의 색감과 디테일이 더욱 뚜렷하게 나타나게 됩니다.

결론

Stable Diffusion에서 ‘가중치’는 프롬프트 내 각 키워드의 중요도를 조절하여 더욱 만족스러운 이미지 생성 결과를 얻을 수 있게 도와주는 유용한 도구입니다. 다양한 가중치 값을 시도해보면서 자신만의 창의적인 이미지를 만들어보세요!