10 Commits

Author	SHA1	Message	Date
Jaret Burkett	338c77d677	Fixed breaking change with diffusers. Allow flowmatch on normal stable diffusion models.	2024-08-22 14:36:22 -06:00
Jaret Burkett	418f5f7e8c	Added new experimental time step weighing that should solve a lot of issues with distribution. Updated example. Removed a warning	2024-08-13 12:02:11 -06:00
Jaret Burkett	9ee1ef2a0a	Added experimental modified sigma sqrt weight mapping for linear timestep scheduling for flowmatching	2024-08-12 17:03:09 -06:00
Jaret Burkett	a6aa4b2c7d	Added ability to set timesteps to linear for flowmatching schedule	2024-08-11 13:06:08 -06:00
Jaret Burkett	fbed8568fb	Actually use the correct timestep sampling instead of calculating it and moving on lol. Tested a few with it and it seems to work better.	2024-08-11 11:10:37 -06:00
Jaret Burkett	e69a520616	Reworked timestep distribution on flowmatch sampler when training.	2024-08-08 06:01:45 -06:00
Jaret Burkett	9beea1c268	Flux training should work now... maybe	2024-08-03 09:17:34 -06:00
Jaret Burkett	bd10d2d668	Some work on sd3 training. Not working	2024-06-13 12:19:16 -06:00
Jaret Burkett	fbec68681d	Added timestep modifications to lcm scheduler for more evenly spaced timesteps	2023-11-17 23:26:52 -07:00
Jaret Burkett	ad50921c41	Sampling tests and added fixes for cleanups	2023-11-16 08:33:23 -07:00