| 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") | |