Jaret Burkett
5ddc5f8ca7
Optimize quantization path
2026-08-28 11:47:29 -06:00
hitoshic1982
5fc8fd76de
Fix cache_text_encoder device preset missing in BaseModel
...
BaseModel.set_device_state_preset() has no branch for the
'cache_text_encoder' preset, so when TextEmbeddingCachingMixin calls
set_device_state_preset('cache_text_encoder') the active_modules list
stays empty and the text encoder is placed on CPU. With
cache_text_embeddings enabled on a BaseModel arch (e.g. chroma), every
caption is then encoded on CPU — with a quantized T5-XXL this took
~6.5 minutes per caption (~21 hours for a 205-image dataset) while the
GPU sat idle. StableDiffusion.set_device_state_preset() already has
this branch; this adds the same two lines to BaseModel.
2026-08-29 01:37:39 +08:00
Jaret Burkett
ce3df32101
Fix speed on wan conv patch
2026-08-28 10:54:26 -06:00
Jaret Burkett
92df289931
Fix quantization issues
2026-08-28 10:48:34 -06:00
Jaret Burkett
85a6880643
Move the offload and quantization to the base model
2026-08-28 08:37:56 -06:00
Jaret Burkett
702254688d
Added legacy paths
2026-08-28 07:55:19 -06:00
Jaret Burkett
c3bc8b0b4e
Full test of ui arch compatability
2026-08-28 06:56:05 -06:00
Jaret Burkett
520d96aac3
Phase 2
2026-08-27 18:18:31 -06:00
Jaret Burkett
45886f01b2
Phase 2
2026-08-27 16:16:35 -06:00
Jaret Burkett
9113420b61
Phase 2
2026-08-27 15:37:13 -06:00
Jaret Burkett
9ed2e0b8e7
Test model loading
2026-08-27 14:31:53 -06:00
Jaret Burkett
8db198ec0a
Phase 1
2026-08-27 11:53:08 -06:00
Jaret Burkett
e8d9cf6d35
Models v2 - phase 0
2026-08-27 10:51:43 -06:00
Jaret Burkett
5497a001cb
Fix issue where errors on qwen image omni were not being fully surfaced.
2026-08-26 20:54:24 -06:00
Jaret Burkett
da79ebce99
Add D-OPSD as a distillation handeling option for MiniMax H3 ref2va
2026-08-26 11:06:33 -06:00
Jaret Burkett
8a912564ce
Fix video DOP methods
2026-08-26 10:05:14 -06:00
Jaret Burkett
8436c407f6
Add ability to delete the loss logs from a selected range on the loss graph
2026-08-23 09:57:47 -06:00
Jaret Burkett
27a03a91f2
Rework progress bar and ui speed string so it gets the same value at a more accurate and snappier wall clock time
2026-08-21 13:31:02 -06:00
Jaret Burkett
b96476a841
Allow overriding the batch size on a per dataset level
2026-08-21 12:56:04 -06:00
Jaret Burkett
89102f76dc
Allow user to pick if reference images are sent in the image or video reference stream for Hidream H3 ref2va
2026-08-19 13:32:06 -06:00
Jaret Burkett
afd1d92722
Fix issue with double sampling qwen vl video frames for hidream h3 for reference videos
2026-08-17 15:39:07 -06:00
Jaret Burkett
42dfe9c661
Improve video frame reader
2026-08-16 10:21:11 -06:00
Jaret Burkett
b982a03ae4
Add training adapter for MiniMax H3 ref2va and default to contrastive loss with adapter
2026-08-16 09:11:07 -06:00
Jaret Burkett
2042481914
On MiniMax H3 set the default to use both contrastive guidance and the training adapter as it is yielding better results than one or the other.
2026-08-16 08:43:31 -06:00
Jaret Burkett
61310c6397
Inprove cpu data acuisition on backend
2026-08-16 08:42:36 -06:00
Jaret Burkett
2cbc2bb097
On minimax h3, trim cached text embeddings to max tokens if they are longer than specified
2026-08-16 08:42:01 -06:00
Jaret Burkett
0f788923ae
Animate action bar so starting a job has a 'working' indicator.
2026-08-15 10:33:40 -06:00
Jaret Burkett
e6cffbc002
Adjust encoded paths so filenames resolve as a basename properly when downloading with things like wget.
2026-08-15 08:56:55 -06:00
Jaret Burkett
151ad0e959
Adjust MiniMah h3 sizing for videos to downscale to match
2026-08-15 07:50:42 -06:00
Jaret Burkett
70b1089359
Version bump
2026-08-15 07:14:13 -06:00
Jaret Burkett
127d6f626d
Rework img/video reference in Minimax H3 to more closely match the comfy ui implementation.
2026-08-15 07:13:54 -06:00
Jaret Burkett
f1faa7725b
version bump
2026-08-15 06:20:32 -06:00
Jaret Burkett
97bf49edad
Add support for video references in MiniMax H3 ref2va
2026-08-15 06:18:09 -06:00
Jaret Burkett
4900e5e866
Version bump
2026-08-14 19:54:51 -06:00
Jaret Burkett
5f53ecde54
Make the viewer thumbnails pul the thumbs
2026-08-14 18:14:12 -06:00
Jaret Burkett
247cb45c3e
Improve gpu cpu monitor efficiency.
2026-08-14 17:14:00 -06:00
Jaret Burkett
695b0baccf
Switch to a live always active gpu and cpu device monitor that can be connected to with an SSE connection for real time device information stream. Much more efficient and faster than previous method of polling.
2026-08-14 12:40:01 -06:00
Jaret Burkett
5261d3fcca
Add checkpointing to the Wan2.1 encoder
2026-08-14 07:31:16 -06:00
Jaret Burkett
0e4b6e8695
Version bump
2026-08-13 12:36:34 -06:00
Jaret Burkett
6ea281973d
Add support for MiniMax H3 Ref2Vid training
2026-08-13 12:36:03 -06:00
Jaret Burkett
ab18528fdb
Let omni captioner handle just images as well. Set it as the new default captioner.
2026-08-13 10:30:17 -06:00
Jaret Burkett
6b7fb60a22
Make contrastive guidance loss do constant instead of sigma loss schedule by default
2026-08-13 09:52:02 -06:00
Jaret Burkett
4e91fb2d0a
Add thinking and abliterated versions of qwen omni.
2026-08-13 08:56:31 -06:00
Jaret Burkett
6c88e3d138
Add layer offloading to omni captioner
2026-08-13 07:21:54 -06:00
Jaret Burkett
a69f3e8710
Version bump
2026-08-12 21:38:00 -06:00
Jaret Burkett
742a4c8cef
Added a Download Full Dataset and Download Captions option to the datasets page so they captions or dataset can be easily downloaded.
2026-08-12 21:37:35 -06:00
Jaret Burkett
81adcc2176
Add caption prompt template picker to the ui. A few styling improvements thrown in for a good time.
2026-08-12 21:23:28 -06:00
Jaret Burkett
ca42a72f4c
Fixed issue with flash attention on omni captioner
2026-08-12 21:14:23 -06:00
Jaret Burkett
7f9a142dfd
Generate thumbnails for dataset items so things load faster and the ui is more stable. Especially good for videos.
2026-08-12 20:33:42 -06:00
Jaret Burkett
175cc1e151
Add Qwen 3 Omni for captioning videos with sound.
2026-08-12 19:57:01 -06:00
Jaret Burkett
hitoshic1982