943 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
Jaret Burkett	bb6db3d635	Added support for caching text embeddings. This is just initial support and will probably fail for some models. Still needs to be ompimized	2025-08-07 10:27:55 -06:00
Jaret Burkett	4c4a10d439	Remove vision model from qwen text encoder as it is not needed for image generation currently	2025-08-06 11:40:02 -06:00
Jaret Burkett	14ccf2f3ce	Refactor qwen5b model code to be qwen 5b specific	2025-08-06 10:54:56 -06:00
Jaret Burkett	5d8922fca2	Add ability to designate a dataset as i2v or t2v for models that support it	2025-08-06 09:29:47 -06:00
Jaret Burkett	1755e58dd9	Update generation script to handle latest models.	2025-08-05 08:55:16 -06:00
Jaret Burkett	6bb3aed9a2	Merge pull request #359 from ostris/qwen_image ... Add support for Qwen Image	2025-08-04 15:51:01 -06:00
Jaret Burkett	74b4d2d291	Version bump	2025-08-04 15:49:32 -06:00
Jaret Burkett	23327d5659	Add qwen image to the ui	2025-08-04 15:48:51 -06:00
Jaret Burkett	93202c7a2b	Training working for Qwen Image	2025-08-04 21:14:30 +00:00
Jaret Burkett	9da8b5408e	Initial but untested support for qwen_image	2025-08-04 13:29:37 -06:00
Jaret Burkett	9dfb614755	Initial work for training wan first and last frame	2025-08-04 11:37:26 -06:00
Jaret Burkett	ef1d60ba34	Update wan 2.2 5b timestep distribution to weighted.	2025-07-30 10:13:22 -06:00
Jaret Burkett	75f688766d	Version bump	2025-07-29 09:30:54 -06:00
Jaret Burkett	a558d5b68f	Move transformer back to device on aggresive wan 2.2 pipeline after generation.	2025-07-29 09:13:47 -06:00
Jaret Burkett	1d1199b15b	Fix bug that prevented training wan 2.2 with batch size greater than 1	2025-07-29 09:06:25 -06:00
Jaret Burkett	f453e28ea3	Fixed deprecation of lumina pipeline error	2025-07-29 08:26:51 -06:00
Jaret Burkett	ca7c5c950b	Add support for Wan2.2 5B	2025-07-29 05:31:54 -06:00
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