Stable Diffusion NSFW
А вот грузим мы, допустим, модельку через 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 😉