Diffusers How to use renderartist/floating-heads-hidream with Diffusers:
pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("HiDream-ai/HiDream-I1-Full", dtype=torch.bfloat16, device_map="cuda")
pipe.load_lora_weights("renderartist/floating-heads-hidream")
prompt = "h3adfl0at 3D floating head of a an old latino man wearing a LA Dodgers baseball cap hat and thick rimmed brown sunglasses with light tinted lenses, he has a thick mustache and looks brooding"
image = pipe(prompt).images[0]