Stable Diffusion NSFW

Артём Агасиев — Thu, 16 Feb 2023 00:59:36 +0300

А вот грузим мы, допустим, модельку через StableDiffusionPipeline, чтобы погенерить всякий полезный в хозяйстве цифровой контент, вбиваем промт и негатив промт, запускаем пайплайн, а на выходе получаем неутешительный черный квадрат Малевича и ошибку вида:


Potential NSFW content was detected in one or more images. A black image will be returned instead. Try again with a different prompt and/or seed.

Чтож, пофиксить такое на самом деле не сложно, нужен всего лишь старый советский... Так, о чем это я? Ага, так вот, чтобы открыть этот ящик Пандоры нужно заменить safety_checker у пайплайна таким вот образом:


pipe.safety_checker = lambda images, clip_input: (images, False)

Sapienti, как говорится, sat 😉

Блог Артёма Агасиева: заметки с тегом NSFW

Stable Diffusion NSFW