import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("fill-in-base-model", dtype=torch.bfloat16, device_map="cuda")
pipe.load_lora_weights("fwwrsd/zimage-lora-4234-13433877")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]
Z Image LoRA — 4234
A personalized LoRA trained with Z Image pipeline for generating images with a specific identity.
Quick Start
Direct Download URL
https://huggingface.co/fwwrsd/zimage-lora-4234-13433877/resolve/main/lora.safetensors
ComfyUI Setup
- Download
lora.safetensors→ place inComfyUI/models/loras/ - Use LoRA Loader node
- Set trigger word:
4234
Download via Command Line
wget https://huggingface.co/fwwrsd/zimage-lora-4234-13433877/resolve/main/lora.safetensors -O lora_4234.safetensors
huggingface-cli download fwwrsd/zimage-lora-4234-13433877 lora.safetensors
Recommended Settings
| Parameter | Value |
|---|---|
| LoRA Strength | 0.7 — 1.0 |
| CFG Scale | 3.5 |
| Steps | 20 — 30 |
Trigger word: 4234 — include in your prompt to activate the LoRA.
Available Checkpoints
This repo contains 5 epoch checkpoints — try different epochs to find the best balance of identity fidelity vs flexibility.
| Epoch | File | Note |
|---|---|---|
| 4 | lora_epoch4.safetensors |
|
| 8 | lora_epoch8.safetensors |
|
| 12 | lora_epoch12.safetensors |
|
| 16 | lora_epoch16.safetensors |
|
| 17 | lora_epoch17.safetensors |
⭐ Best (recommended) |
lora.safetensors is a copy of the best (final) checkpoint — epoch 17.
Download Specific Epoch
# Download epoch 17 (best)
huggingface-cli download fwwrsd/zimage-lora-4234-13433877 lora_epoch17.safetensors
# Download any other epoch
huggingface-cli download fwwrsd/zimage-lora-4234-13433877 lora_epoch<N>.safetensors
Training Details
| Parameter | Value |
|---|---|
| Training Method | diffusion-pipe + DeepSpeed |
| LoRA Rank | 16 |
| Learning Rate | 1e-4 |
| Total Epochs | unknown (best: 17) |
| Resolution | 1024px |
| Dataset Size | 58 images |
| Preset | test |
| Created | 2026-03-04 |
License
Apache 2.0 — free for personal and commercial use.
Trained with NanoBanana LoRA Bot on RunPod
- Downloads last month
- 50