This is a fine-tuned Stable Diffusion XL (SDXL) model based on
FLUX.1-dev. It's designed to generate high-quality images with a focus on
[specific theme or style your model specializes in].
1from diffusers import StableDiffusionXLPipeline
2import torch
3
4model_id = "colt12/maxcushion"
5pipe = StableDiffusionXLPipeline.from_pretrained(
6 model_id, torch_dtype=torch.float16, use_safetensors=True, variant="fp16"
7)
8pipe = pipe.to("cuda")
9
10prompt = "Your prompt here"
11image = pipe(prompt).images[0]
12image.save("generated_image.png")