게시글중에서 waifu-diffusion 사용법을 물어보는 글이 있길래 간단하게 써봄

waif-diffusion은 Hugging Space에 모델이 올려져있기 때문에 colab에서 불러와서 바로 실행이 가능함


1. https://huggingface.co/hakurei/waifu-diffusion 사이트로 이동


a16711aa372a07f720afd8b236ef203e3ff1d281f39450

2. 밑쪽에 있는 "Open in Colab" 버튼 클릭


a16711aa372a07f420afd8b236ef203e2daededa69ade6

3. 위의 메뉴에서 "Edit" > "Notebook settings" 클릭


a16711aa372a07f520afd8b236ef203ebe5e6502beeefc

4. "Hardware accelerator"를 "GPU"로 설정하고 "Save" 버튼을 누른다.


a16711aa372a07f320afd8b236ef203ea8b9c9f27044

5. 왼쪽에 실행버튼 두개를 누른다.


a16711aa372a07f220afd8b236ef203e2beaeba5ca634cbe

6. 그러면 밑쪽에 작은 입력창이 뜨는데 여기서 원하는 글을 입력하고 "Run" 버튼을 누르면 된다.


+ 추가 팁

1. 한번에 띄우는 이미지 개수 조절

a16711aa372a07f020afd8b236ef203e7569352f3810

num_samples의 값을 조절하면 한번에 생성하는 이미지의 개수를 조정할 수 있음. 하지만 메모리용량때문에 최대 4개 정도가 한계인듯


2. NSFW 필터 제거

a16711aa372a07f120afd8b236ef203ec089ff95ce99

일반적으로 수위가 너무 높은 이미지일경우 파이프라인에서 자동으로 경고문을 띄우면서 검정색 이미지로 대체되는데 만약 이러한 필터링을 제거하고 싶을 경우,


a16711aa372a07fe20afd8b236ef203e8da67bc9ac3f

pipe... 소스코드 밑에 다음과 같이 필터를 제거하는 소스코드를 추가하고, 재실행시키면 됨


--- 소스코드

def dummy(images, **kwargs):
return images, False
pipe.safety_checker = dummy