Hellcoder - (Trying to uncensor Stable Diffusion 2.1)

2.1 모델과 그 파생형을 학습시키는 것이 얼마나 어려웠는지 눈치챘나요? 그 이유는 텍스트 인코더가 심하게 검열되어 있었기 때문입니다. 이 텍스트 인코더는 이미지와 쌍을 이루며, 모델에 제공한 이미지 내부 내용을 대략적으로 이해하는 역할을 합니다.

좋고 검열되지 않은 텍스트 인코더 중 2.0-2.1 모델과 호환되는 것들이 HuggingFace에 존재합니다. 원하는 모델을 Diffusers 형식으로 얻은 후, 모델의 단일 구성 요소를 완전히 교체할 수 있습니다. 제가 이 모델에 대해 바로 이 작업을 수행했으며, 앞으로도 계속 개선할 계획입니다.

저는 waifu diffusion 태거를 사용하여 다양한 여성들을 대상으로 큰 양의 데이터로 모델을 학습시켰고, 이를 통해 모델은 각 여성의 개별 특징을 포착하고 이를 결합하여 인코더는 알고 있지만 U-Net은 학습되지 않은 유명인사들의 정보를 도출해냈습니다.

현재 이 모델은 NSFW 콘텐츠에 대한 생성 능력이 낮지만, 다음 단계로 이를 개선해보려 합니다. 그 전까지는 유명인사들의 어느 정도 수용 가능한 이미지를 생성할 수 있습니다. 이 사진들이 완벽하지 않다는 점을 잘 알고 있지만, 기본 모델에 비해 큰 진전입니다.

아, 이 모델은 과적합되지 않았기 때문에 다른 개념에도 대응할 수 있을 것입니다. 직접 시도해보세요. 또한, Illuminutty와 지능적으로 결합되었기 때문에 그들의 임베딩도 시도해보세요...