Creation of Images from Text-Prompt (Customized Training)

Beginners
24

I added the omission of “Safety_checker”, but it is still giving a black image:

pipeline = StableDiffusionPipeline.from_pretrained(
    "stabilityai/stable-diffusion-2-1-base",
    torch_dtype=torch.float16,
    safety_checker=None
).to("cuda")

I am getting the following warning:

D:\Ganu\AIImage\huggingface\kohya_ss\Python310\lib\site-packages\diffusers\image_processor.py:147: RuntimeWarning: invalid value encountered in cast
  images = (images * 255).round().astype("uint8")
1 Like