oraul

/

finetuned_bratpit_comps

like 0

from diffusers import StableDiffusionPipeline
import torch

model = "oraul/Stability_CelebsHAQ_v1"
pipeline = StableDiffusionPipeline.from_pretrained(model_ckpt_tuned, torch_dtype=torch.float16).to("cuda")

prompt = "a yougster boy batch2_proj_11 image" # here, batch2_proj_11 is the placeholder that identifies unique style in the model. please add it into the angel brackets
image = pipeline(prompt_tuned).images[0]