import torch
from diffusers import FluxPipeline
pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16)
pipe.enable_model_cpu_offload()
prompt = "A lot of people takeing a photo of you"
image = pipe(
prompt,
height = 1024,
width = 1024,
guidance_scale = 3.5,
num_inference_steps = 60,
max_sequence_lenth=255,
generator = torch.Generator("cpu").manual_seed(0)
).images(0)
image.save("photo_of_you.png")
This is my code. I'm trying to make an image to test it. When I try to run it I get this output. I posted an image of it.