577 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
Jaret Burkett	7e37918fbc	Double tap module casting as it doesent seem to happen every time.	2025-03-07 22:15:24 -07:00
Jaret Burkett	4d88f8f218	Fixed cuda error when not all tensors have been moved to the correct device.	2025-03-07 22:04:35 -07:00
Jaret Burkett	25341c4613	Got wan 14b training to work on 24GB card.	2025-03-07 17:04:10 -07:00
Jaret Burkett	391cf80fea	Added training for Wan2.1. Not finalized, wait.	2025-03-07 13:53:44 -07:00
Jaret Burkett	4e3bda7c70	Merge pull request #264 from ostris/cogview4 ... Added basics for CogView4. Broken as hell though. Dont use.	2025-03-05 14:52:06 -07:00
Jaret Burkett	763128ea42	Note about cogview	2025-03-05 14:46:11 -07:00
Jaret Burkett	4fe33f51c1	Fix issue with picking layers for quantization, adjust layers fo better quantization of cogview4	2025-03-05 13:44:40 -07:00
Jaret Burkett	aa44828c0c	WIP more work on cogview4	2025-03-05 09:43:00 -07:00
Jaret Burkett	6f6fb90812	Added cogview4. Loss still needs work.	2025-03-04 18:43:52 -07:00
Jaret Burkett	c57434ad7b	Removed wan submodule stuff for now	2025-03-04 00:32:24 -07:00
Jaret Burkett	8bb47d1bfe	Merge branch 'main' into wan21	2025-03-04 00:31:57 -07:00
Jaret Burkett	e7dbb20f68	Removed wan submodule for now	2025-03-04 00:29:19 -07:00
Jaret Burkett	c5e0c2bbe2	Fixes to allow for redux assisted training	2025-03-03 16:27:19 -07:00
Jaret Burkett	1f3f45a48d	Bugfixes	2025-03-03 08:22:15 -07:00
Jaret Burkett	3c8c84f156	Added supporters to readme and a script to update it	2025-03-02 10:25:27 -07:00
Jaret Burkett	b001d77efb	Added LoKr instructions to the readme	2025-03-02 08:55:56 -07:00
Jaret Burkett	7ae31c9ae9	Added LoKr to the ui	2025-03-02 08:49:01 -07:00
Jaret Burkett	b16819f8e7	Added LoKr support	2025-03-02 06:57:50 -07:00
Jaret Burkett	f5e40dfa62	WIP on wan	2025-03-01 16:12:52 -07:00
Jaret Burkett	acc79956aa	WIP create new class to add new models more easily	2025-03-01 13:49:02 -07:00
Jaret Burkett	60539c0b0f	Allow using prior loss with a custom adapter	2025-03-01 08:01:14 -07:00
Jaret Burkett	dd700f70b3	Avoid loading state dict for automagic for now until I can sort out some issues	2025-02-26 17:03:14 -07:00
Jaret Burkett	d360e76661	fixed issue with dop prompt replacement	2025-02-26 13:35:18 -07:00
Jaret Burkett	6ec23ed226	Fixed issue when doing inverted masked prior with flowmatching algos	2025-02-26 12:12:32 -07:00
Jaret Burkett	f6e16e582a	Added Differential Output Preservation Loss to trainer and ui	2025-02-25 20:12:36 -07:00
Jaret Burkett	259ded9602	Fixed issue with trigger word saving in ui	2025-02-24 11:04:24 -07:00
Jaret Burkett	440ba5fb3d	Spawn windows in an cmd terminal. Should be working now, but not sure on my system	2025-02-24 08:54:56 -07:00
Jaret Burkett	093f14ac19	UI Bug fixes and initial windows support	2025-02-24 08:15:22 -07:00
Jaret Burkett	f0fbd8bb53	Merge pull request #256 from ostris/ui ... Added AI-Toolkit UI	2025-02-23 16:10:49 -07:00
Jaret Burkett	0a981bea2b	Fixed typo	2025-02-23 16:07:38 -07:00
Jaret Burkett	1d0e3a4498	Fixed some build issues for now. Added info to the readme	2025-02-23 15:59:17 -07:00
Jaret Burkett	3c7daf49f3	Add HF token to env when spawing via ui	2025-02-23 14:52:12 -07:00
Jaret Burkett	56d8d6bd81	Capture speed from the timer for the ui	2025-02-23 14:38:46 -07:00
Jaret Burkett	3e49337a58	Set step to the last step saved at when exiting	2025-02-23 13:21:22 -07:00
Jaret Burkett	60f848a877	Send more data when loading the model to the ui	2025-02-23 12:49:54 -07:00
Jaret Burkett	b366e46f1c	Added more settings to the training config	2025-02-23 12:34:52 -07:00
Jaret Burkett	a280f78c69	Added checkpoint downloader	2025-02-22 16:48:15 -07:00
Jaret Burkett	6e19e7449e	Fixed some issues with gpu info refreshing	2025-02-22 14:14:23 -07:00
Jaret Burkett	a6d46ad9ae	Cleanup of job page	2025-02-22 13:54:06 -07:00
Jaret Burkett	f3725578dd	Cleaned up dashboard	2025-02-22 13:23:26 -07:00
Jaret Burkett	ed99c3c0c8	Moved gpu to its own widget	2025-02-22 12:43:20 -07:00
Jaret Burkett	ed84c19205	Moved the job action bar to a shred component	2025-02-22 12:20:14 -07:00
Jaret Burkett	a7a9c11d9e	Fixed add image dropbox	2025-02-22 11:59:21 -07:00
Jaret Burkett	f60698d0ee	Fixed some bugs with ui and lock job name to prevent issues with continuing training.	2025-02-22 11:49:36 -07:00
Jaret Burkett	5f094fb17a	Added controls to the jobs table	2025-02-22 10:57:53 -07:00
Jaret Burkett	a5227cba7b	Switched to a universal table library	2025-02-22 09:59:17 -07:00
Jaret Burkett	77a5e01301	Added proper icon	2025-02-22 09:05:55 -07:00
Jaret Burkett	4ef5a668c0	Make left arrow browsing only hit last image max	2025-02-21 22:04:53 -07:00
Jaret Burkett	f081d14527	Preview samples full screen and use arrow keys to navigate them	2025-02-21 21:52:24 -07:00
Jaret Burkett	710c6de1c9	Samples work in ui now	2025-02-21 20:28:52 -07:00