触发词
文件清单 线路:镜像(hf-mirror.com) · 切换
下载后放进对应的 ComfyUI 目录。models/ 指 ComfyUI 的模型根目录。
说明
Fyre (fictional character)
作者:jandordoe
Fyre, a fictional character, redhead with freckles.
Trained on 20 Klein 9b generated images generated from an initial z-image generated image.
H3 image Trained with ai-toolkit, lokr factor 4, automagic v3 LR 0.0005, weight decay 0.00001, Shift/Balanced, EMA 0.98. Training config is in Other Files as config-h3-image-v01.yml
Excellent likeness was achieved by step 2,000, with step 3,000 showing a marginal, but measurable, increase. This model achieves better measured likeness than any other model so far, even at step 2,000.
Samples are not cherry picked - at least two demonstrate some anatomy issues with H3 image.
moss image Trained with ai-toolkit, lokr factor 4, automagic v3 LR 0.0001, weight decay 0.0002, Shift/Balanced. Training config is in Other Files as config-moss-v01.yml
qwen image 2.1 Trained with ai-toolkit, lokr factor 4, automagic v3, LR 0.0005. 1,800 steps outperformed 2,000 steps for facial match. ai-toolkit configuration attached as config-qwen21.toml.zip
minimax h3 Trained with ai-toolkit, automagic v3 with Contrastive Guidance and training adapter v1; differential guidance scale = 1; rank 24. ai-toolkit configuration attached as minimaxh3-fyre-v01.yaml
krea 2 Trained with ai-toolkit, automagic v3 optimizer, LR/weight decay 0.0001. Training images, captions, and ai-toolkit config are included in fyre-krea2-v01.zip
ideogram 4 Trained with ai-toolkit, automagic v2 optimizer, LR 0.0001, rank 32, with captions generated by ai-toolkit (including bounding boxes) with some manual revisions to include the "Fyre" keyword. Training images, captions, and ai-toolkit config are included in fyre-ideogram-v01.zip
z-image / qwen 2512 / wan 2.2 14b low noise
Trained with diffusion-pipe, this is a 128-rank lora trained on z-image (non-turbo, aka "base"), wan 2.2 14B low noise, and qwen image 2512, using the ProdigyPlusScheduleFree optimizer (https://github.com/tdrussell/diffusion-pipe/pull/483), and masked training with backgrounds masked to 10%.
The model was evaluated every 200 steps (10 repeats of each training image) using FaceNet512 to evaluate the face similarity, using the same settings, seed, etc for inference, calculating the average distance to a subset of 10 of the training images. For the z-image model, the best face similarity was achieved around epoch 16 and 17 (3,200 steps and 3,400 steps). For the qwen image 2512 model, the best face similarity was achieved at epoch 20 (4,000 steps). For wan 2.2 low noise, the best face similarity was achieved at epoch 10 (2,000) steps.
The attached training data zip includes: the diffusion pipe config files; the training images, prompts, and masks; the validation images (subset of training images); and - for the z-image model - the output of MirrorMetrics evaluation of epoch16 and 17 vs. the validation set.
z-image model evaluation:
- For average distances, lower is better. Distance under 0.3 is considered a face match.
- Matching percentage is the percent of the 10 validation images that the generated face was under 0.3 distance from.
- epoch 16 (3,200 steps)
- z-image (non-turbo) inference
- Average distance: 0.187
- Matching percentage: 100%
- z-image-turbo inference
- Average distance: 0.262
- Matching percentage: 80%
- epoch 17 (3,400 steps)
- z-image (non-turbo) inference
- Average distance: 0.193
- Matching percentage: 100%
- z-image-turbo inference
- Average distance: 0.255
- Matching percentage: 80%
Evaluation using MirrorMetrics (html output included in training data attachment)
Training config: (included in attached training data attachment)
Training command: PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True NCCL_P2P_DISABLE="1" NCCL_IB_DISABLE="1" deepspeed \ --num_gpus=1 train.py \ --deepspeed \ --regenerate_cache \ --config /home/user/diffusion-pipe-config.toml
