import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("Tongyi-MAI/Z-Image-Turbo", dtype=torch.bfloat16, device_map="cuda")
pipe.load_lora_weights("Jorgegriese/Penis")
prompt = "-"
image = pipe(prompt).images[0]Not-For-All-Audiences
This repository has been marked as containing sensitive content and may contain potentially harmful and sensitive information.
View model card