You can deploy this model and start earning money today!

Grantedkitty073/
dpo-stablediffxl-text2image-v2

Name: dpo-stablediffxl-text2image-v2
Availability: InStock

diffusers

safetensors

Views

No views yet

A quick example

python

from diffusers import StableDiffusionXLPipeline, UNet2DConditionModel
import torch

# load pipeline
model_id = "stabilityai/stable-diffusion-xl-base-1.0"
pipe = StableDiffusionXLPipeline.from_pretrained(model_id, torch_dtype=torch.float16, variant="fp16", use_safetensors=True).to("cuda")

# load finetuned model
unet_id = "Grantedkitty073/dpo-sdxl-text2image-v1"
unet = UNet2DConditionModel.from_pretrained(unet_id, subfolder="unet", torch_dtype=torch.float16)
pipe.unet = unet
pipe = pipe.to("cuda")

prompt = "Cute little dog with human body"
image = pipe(prompt, guidance_scale=5).images[0].resize((512,512))
    
image.save("cutelitledog.png")

Grantedkitty073/dpo-stablediffxl-text2image-v2

A quick example

Grantedkitty073/
dpo-stablediffxl-text2image-v2