90 Commits

Author	SHA1	Message	Date
Hameer Abbasi	5e86139e0a	Fix NameError.	2025-06-11 15:07:20 +02:00
Hameer Abbasi	c5d6b74fea	Fix caption loading.	2025-06-11 15:05:31 +02:00
Jaret Burkett	22cdfadab6	Added new timestep weighing strategy	2025-06-04 01:16:02 -06:00
Jaret Burkett	1210050ead	Reworked control generator. It is now significantly faster. Also uses better pose model with better license.	2025-05-08 14:35:55 -06:00
Jaret Burkett	c12036df95	Added ability to use short captions from json caption file	2025-04-16 08:32:28 -06:00
Jaret Burkett	d8bdc03256	Allow full control of caption extensions	2025-04-10 07:42:04 -06:00
Jaret Burkett	96ba2fd129	Added methods to the dataloader to automatically generate controls for line, mask, inpainting, depth, and pose.	2025-04-09 13:35:04 -06:00
Jaret Burkett	1d5f387f54	Fix docker command to work better with runpod	2025-03-27 17:44:46 -06:00
Jaret Burkett	45be82d5d6	Handle inpainting training for control_lora adapter	2025-03-24 13:17:47 -06:00
Jaret Burkett	f10937e6da	Handle multi control inputs for control lora training	2025-03-23 07:37:08 -06:00
Jaret Burkett	b829983b16	Added ability to load video datasets and train with them	2025-03-19 09:54:26 -06:00
Jaret Burkett	c52421aab7	Allow clip image to not have processor on dataloader for raw img	2025-03-15 08:27:54 -06:00
Jaret Burkett	5e663746b8	Working multi gpu training. Still need a lot of tweaks and testing.	2025-01-25 16:46:20 -07:00
Jaret Burkett	3922981996	Added some additional experimental things to the vision direct encoder	2024-10-10 19:42:26 +00:00
Jaret Burkett	ab22674980	Allow for a default caption file in the folder. Minor bug fixes.	2024-10-10 07:31:33 -06:00
Jaret Burkett	28e6f00790	Fixed bug in returning clip image embed to actually return it	2024-10-03 10:49:09 -06:00
Jaret Burkett	67e0aca750	Added ability to load clip pairs randomly from folder. Other small bug fixes	2024-10-03 10:03:49 -06:00
Jaret Burkett	58537fc92b	Added initial direct vision pixtral support	2024-09-28 10:47:51 -06:00
liaoliaojun	e127c079da	fix: print out the path where the image encode failed (#107)	2024-08-22 21:34:35 -06:00
Jaret Burkett	9001e5c933	Change flux latent spact if so it will not use old cache	2024-08-14 11:27:40 -06:00
Jaret Burkett	87ba867fdc	Added flux training. Still a WIP. Wont train right without rectified flow working right	2024-08-02 15:00:30 -06:00
Jaret Burkett	03613c523f	Bugfixes and cleanup	2024-08-01 11:45:12 -06:00
Jaret Burkett	80aa2dbb80	New image generation img2img. various tweaks and fixes	2024-07-24 04:13:41 -06:00
Jaret Burkett	e4558dff4b	Partial implementation for training auraflow.	2024-07-12 12:11:38 -06:00
Jaret Burkett	bd10d2d668	Some work on sd3 training. Not working	2024-06-13 12:19:16 -06:00
Jaret Burkett	3f3636b788	Bug fixes and little improvements here and there.	2024-06-08 06:24:20 -06:00
Jaret Burkett	833c833f28	WIP on SAFE encoder. Work on fp16 training improvements. Various other tweaks and improvements	2024-05-27 10:50:24 -06:00
Jaret Burkett	10e1ecf1e8	Added single value adapter training	2024-04-28 06:04:47 -06:00
Jaret Burkett	b96913d73c	Improvements to dataloader	2024-04-27 09:28:28 -06:00
Jaret Burkett	5da3613e0b	Bug fixes and minor features	2024-04-25 06:14:31 -06:00
Jaret Burkett	427847ac4c	Small tweaks and fixes for specialized ip adapter training	2024-03-26 11:35:26 -06:00
Jaret Burkett	9c1cc9641e	Added keep tokens to keep so many tokens in a prompt when dropping	2024-03-18 13:18:25 -06:00
Jaret Burkett	337945de9a	Added this not that guidance. Added ability to replace prompts.	2024-02-28 20:10:14 -07:00
Jaret Burkett	f965a1299f	Fixed Dora implementation. Still highly experimental	2024-02-24 10:26:01 -07:00
Jaret Burkett	4ec4025cbb	Added adapter modules for text encoders and direct vision	2024-02-12 08:46:18 -07:00
Jaret Burkett	1ae1017748	Bug fixes. added ability to use l1 loss. varous other tests and improvements	2024-01-31 06:30:54 -07:00
Jaret Burkett	92b9c71d44	Many bug fixes. Ip adapter bug fixes. Added noise to unconditional, it works better. added an ilora adapter for 1 shotting LoRAs	2024-01-28 08:20:03 -07:00
Jaret Burkett	f17ad8d794	various bug fixes. Created an contextual alpha mask module to calculate alpha mask	2024-01-18 16:34:27 -07:00
Jaret Burkett	5276975fb0	Added additional config options for custom plugins I needed	2024-01-15 08:31:09 -07:00
Jaret Burkett	290393f7ae	Imporvements to ip weight adaptation. Bug fixes. Added masking to direct guidance loss. Allow importing a file for random triggers. Handle bas meta images with improper sizing.	2024-01-11 12:22:16 -07:00
Jaret Burkett	b2a54c8f36	Added siglip support	2024-01-09 20:52:21 -07:00
Jaret Burkett	645b27f97a	Bug fixes with ip adapter training. Made a clip pre processor that can be trained with ip adapter to help augment the clip input to squeeze in more detail from a larget input. moved clip processing to the dataloader for speed.	2024-01-04 12:59:38 -07:00
Jaret Burkett	d11ed7f66c	Big fixes and added method to standardize values in both latent and pixel space before feeding into the network. Target values were determined over huge generated regularization sets.	2023-12-26 06:19:48 -07:00
Jaret Burkett	0f8daa5612	Bug fixes, work on maing IP adapters more customizable.	2023-12-24 08:32:39 -07:00
Jaret Burkett	7703e3a15e	Fixes for sdxl ip adapter training. Bug fixes	2023-12-21 11:15:58 -07:00
Jaret Burkett	0f597f453e	Switched ip adapter dataloader to clip_image paths so the control paths can be used for training assistant adapters while training ip adapters	2023-12-20 10:32:24 -07:00
Jaret Burkett	82098e5d6e	Added more functionality for ip adapters	2023-12-19 09:54:56 -07:00
Jaret Burkett	b653906715	Fixed ip adapter training. Works now	2023-12-17 08:22:59 -07:00
Jaret Burkett	792a5e37e2	Numerous fixes for time sampling. Still not perfect	2023-11-28 07:34:43 -07:00
Jaret Burkett	e47006ed70	Added some features for an LCM condenser plugin	2023-11-15 08:56:45 -07:00