Qwen-Image-2.1-viewpoint-orbit-LoRA / train_lora_qwen21_orbit.yaml
ysharma's picture
ysharma HF Staff
Add ai-toolkit training config: qwen_image_2 RGBA edit LoRA rank 32
cec47fc verified
Raw History Blame Contribute Delete
1.47 kB
job: extension
config:
name: "orbit_alpha_qwen21"
process:
- type: "sd_trainer"
training_folder: "/output"
device: cuda:0
performance_log_every: 100
network:
type: "lora"
linear: 32
linear_alpha: 32
save:
dtype: "float16"
save_every: 500
max_step_saves_to_keep: 8
push_to_hub: false
logging:
use_ui_logger: true
log_every: 1
datasets:
- folder_path: "/train_data/targets"
control_path: "/train_data/controls"
caption_ext: "txt"
resolution: [768]
cache_latents_to_disk: true
train:
batch_size: 1
steps: 2000
gradient_accumulation: 1
timestep_type: "weighted"
train_unet: true
train_text_encoder: false
gradient_checkpointing: true
noise_scheduler: "flowmatch"
optimizer: "adamw8bit"
lr: 1e-4
dtype: bf16
cache_text_embeddings: true
skip_first_sample: true
disable_sampling: true
model:
name_or_path: "Qwen/Qwen-Image-2.1"
arch: "qwen_image_2"
quantize: false
quantize_te: false
model_kwargs:
rgba: true
match_target_res: true
meta:
name: "[name]"
version: "1.0"
description: "RGBA viewpoint-orbit LoRA for Qwen-Image-2.1; relative-azimuth orbit grammar; trained on Google Scanned Objects renders (CC-BY 4.0)"