Views
No views yet
bs=8/rank x 2 A40s)to_q, to_k, to_v, to_out.0,
feed_forward.w1/w2/w3ZImagePipeline, 50 steps, guidance_scale=5.0t = (1000 - sched.timesteps[idx]) / 1000
matching HF Diffusers'
examples/dreambooth/train_dreambooth_lora_z_image.py.
The flow-matching target is equivalent to noise - model_input, with the
Z-Image pipeline pre-step negation handled by model_pred = -model_pred.1import torch
2from diffusers import ZImagePipeline
3
4pipe = ZImagePipeline.from_pretrained(
5 "Tongyi-MAI/Z-Image", torch_dtype=torch.bfloat16
6).to("cuda")
7pipe.load_lora_weights("PNW-GM/z-image-witcher3-velen-lora")
8
9image = pipe(
10 prompt="aerial cartographic view of a marshland village in the Velen style",
11 num_inference_steps=50,
12 guidance_scale=5.0,
13).images[0]
14image.save("velen.png")