import os
import torch
from diffusers import FluxPipeline

# Force CPU
os.environ["CUDA_VISIBLE_DEVICES"] = ""
os.environ["DIFFUSERS_NO_BITSANDBYTES"] = "1"

pipe = FluxPipeline.from_pretrained(
    "black-forest-labs/FLUX.1-dev",
    torch_dtype=torch.float32
)

pipe.to("cpu")

image = pipe(
    prompt="a cinematic portrait of a cyberpunk warrior",
    num_inference_steps=20
).images[0]

image.save("output.png")
print("✅ Generated on CPU")