Stable Diffusion 샘플링 스크립트

Stable Diffusion v1의 공식 저장소는 텍스트-이미지 생성을 위한 참고 샘플링 스크립트 scripts/txt2img.py를 제공해요. 추론 설정을 커맨드 라인 인자로 조절할 수 있죠.

출처: Stable Diffusion README

Stable Diffusion v1은 (비풀링된) CLIP ViT-L/14 텍스트 인코더의 텍스트 임베딩으로 조건화된 잠재 디퓨전 모델이에요. 공식 저장소는 샘플링용 참조 스크립트를 제공하고, 더 활발한 커뮤니티 개발이 이뤄지는 곳은 diffusers 통합이에요.

참조 샘플링 스크립트는 두 가지 안전장치를 포함해요.

Safety Checker Module — 노골적 출력의 확률을 줄이는 역할
인비저블 워터마킹 — 출력 이미지가 머신 생성물임을 식별하도록 도움

scripts/txt2img.py는 PLMS 샘플러를 구현했고, 학습된 해상도인 512x512 크기의 이미지를 50 스텝으로 렌더링해요. 지원되는 인자 전체는 python scripts/txt2img.py --help로 확인할 수 있어요. 주요 인자 일부를 보면 이래요.

python scripts/txt2img.py --prompt "a cat" --plms --seed 42

--prompt [PROMPT] — 생성할 텍스트 프롬프트
--plms — PLMS 샘플러 사용 (기본은 DDIM)
--ddim_steps DDIM_STEPS / --ddim_eta DDIM_ETA — 샘플링 스텝·eta 조절
--n_iter N_ITER / --n_samples N_SAMPLES — 반복·샘플 수
--H H / --W W — 생성 이미지 높이·너비
--scale SCALE — CFG(무분류자 가이드) 스케일
--seed SEED — 랜덤 시드
--precision {full,autocast} — 연산 정밀도
--ckpt CKPT / --config CONFIG — 체크포인트·설정 파일 경로

이런 인자로 프롬프트, 해상도, 시드, 샘플링 스텝을 바꿔가며 생성 결과를 다양하게 뽑아볼 수 있어요.