776 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
Jaret Burkett	e55116d8c9	Added hidream low vram options	2025-07-27 18:29:46 -06:00
Jaret Burkett	99705ec8be	Add support in UI for Hidream E1	2025-07-27 18:13:36 -06:00
Jaret Burkett	ed8d14225f	Add ability to set the quantization type for text encoders and transformer in the ui	2025-07-27 18:00:53 -06:00
Jaret Burkett	b717586ee2	Version bump	2025-07-27 15:13:28 -06:00
Jaret Burkett	cefa2ca5fe	Added initial support for Hidream E1 training	2025-07-27 15:12:56 -06:00
Jaret Burkett	3f518d9951	Add sharpening before losses with a split loss on vae training	2025-07-27 15:11:56 -06:00
Jaret Burkett	77dc38a574	Some work on caching text embeddings	2025-07-26 09:22:04 -06:00
Jaret Burkett	0d89c44624	Bug fixes on vae trainer. Allow to target params for vae training.	2025-07-26 09:20:22 -06:00
Jaret Burkett	3e14a674ac	Fix upload progress for datasets in the ui	2025-07-26 09:07:30 -06:00
Jaret Burkett	523c159579	Add vram flag to some models in the ui	2025-07-24 07:02:46 -06:00
Jaret Burkett	c5eb763342	Improvements to VAE trainer. Allow CLIP loss.	2025-07-24 06:50:56 -06:00
Jaret Burkett	ca5cf827a1	Version bump	2025-07-20 12:20:46 -06:00
Jaret Burkett	b1bff66d52	Merge pull request #343 from davertor/fix_kontext_bs ... fix: Guidance incorrect shape	2025-07-20 12:00:55 -06:00
Daniel Verdu	a77ba5a089	fix: Guidance incorrect shape	2025-07-18 12:49:18 +02:00
Jaret Burkett	8610c6ed7f	Made it easy to add control images to the samples in the UI	2025-07-17 12:00:48 -06:00
Jaret Burkett	e25d2feddf	Use scale shift in vae latent space for vae trainer	2025-07-17 08:14:07 -06:00
Jaret Burkett	f500b9f240	Add ability to do more advanced sample prompt objects to prepart for a UI rework on control images and other things.	2025-07-17 07:13:35 -06:00
Jaret Burkett	3916e67455	Scale target vae latent before targeting it	2025-07-17 07:12:21 -06:00
Jaret Burkett	e5ed450dc7	Allow finetuning tiny autoencoder in vae trainer	2025-07-16 07:13:30 -06:00
Jaret Burkett	1930c3edea	Fix naming with wan i2v new keys in lora	2025-07-14 07:34:01 -06:00
Jaret Burkett	ef5149180c	Switch i2v ui defaults to weighted	2025-07-12 21:30:04 -06:00
Jaret Burkett	998e8b6537	Bump version	2025-07-12 16:57:09 -06:00
Jaret Burkett	755f0e207c	Fix issue with wan i2v scaling. Adjust aggressive loader to be compatable with updated diffusers.	2025-07-12 16:56:27 -06:00
Jaret Burkett	2e84b3d5b1	Update VAE trainer to handle fixed latent target. Also minor bug fixes and improvements	2025-07-12 16:55:15 -06:00
Jaret Burkett	7ab44ae0cd	Fix issue with getting captions on runpod	2025-07-11 19:16:50 +00:00
Jaret Burkett	47002b067f	Add path to image for datasets on the image card	2025-07-11 11:44:48 -06:00
Jaret Burkett	8537a8557f	Add simple ui settings to train Wan i2v models.	2025-07-11 11:28:40 -06:00
Jaret Burkett	6e2beef8dd	Version Bump	2025-07-09 13:55:33 -06:00
Jaret Burkett	611969ec1f	Allow control image for omnigen training and sampling	2025-07-09 13:54:55 -06:00
Jaret Burkett	bbb57de6ec	Speed up omnigen TE loading	2025-07-05 09:32:00 -06:00
Jaret Burkett	5906a76666	Fixed issue with flux kontext forcing generation image sizes	2025-06-29 05:38:20 -06:00
Jaret Burkett	57a81bc0db	Update base model version for kontext meta	2025-06-28 14:48:36 -06:00
Jaret Burkett	843be31138	Update readme changelog	2025-06-28 12:55:23 -06:00
Jaret Burkett	8fb01e96e4	Update sponsors	2025-06-28 10:05:17 -06:00
Jaret Burkett	01a3c8a9b1	Fix device issue	2025-06-26 19:14:25 -06:00
Jaret Burkett	4f91cb7148	Fix issue with gradient checkpointing and flux kontext	2025-06-26 19:03:12 -06:00
Jaret Burkett	446b0b6989	Remove revision for kontext	2025-06-26 16:46:58 -06:00
Jaret Burkett	60ef2f1df7	Added support for FLUX.1-Kontext-dev	2025-06-26 15:24:37 -06:00
Jaret Burkett	8d9c47316a	Work on mean flow. Minor bug fixes. Omnigen improvements	2025-06-26 13:46:20 -06:00
Jaret Burkett	84c6edca7e	Merge branch 'main' into dev	2025-06-25 14:10:25 -06:00
Jaret Burkett	24cd94929e	Fix bug that can happen with fast processing dataset	2025-06-25 14:01:08 -06:00
Jaret Burkett	19ea8ecc38	Added support for finetuning OmniGen2.	2025-06-25 13:58:16 -06:00
Jaret Burkett	18513ec866	Merged in from main	2025-06-24 10:56:54 -06:00
Jaret Burkett	5e733764aa	Update version	2025-06-24 10:37:13 -06:00
Jaret Burkett	03bc431279	Fixed an issue training lumina 2	2025-06-24 10:29:47 -06:00
Jaret Burkett	f3eb1dff42	Add a config flag to trigger fast image size db builder. Add config flag to set unconditional prompt for guidance loss	2025-06-24 08:51:29 -06:00
Jaret Burkett	ba1274d99e	Added a guidance burning loss. Modified DFE to work with new model. Bug fixes	2025-06-23 08:38:27 -06:00
Jaret Burkett	8602470952	Updated diffusion feature extractor	2025-06-19 15:36:10 -06:00
Jaret Burkett	4586eb5392	Added social links to sidebar	2025-06-17 13:25:24 -06:00
Jaret Burkett	989ebfaa11	Added a basic torch profiler that can be used in config during development to find some obvious issues.	2025-06-17 13:03:39 -06:00