Plat 79b4e04b80 Feat: Wandb logging (#95) ... * wandb logging * fix: start logging before train loop * chore: add wandb dir to gitignore * fix: wrap wandb functions * fix: forget to send last samples * chore: use valid type * chore: use None when not type-checking * chore: resolved complicated logic * fix: follow log_every --------- Co-authored-by: Plat <github@p1at.dev> Co-authored-by: Jaret Burkett <jaretburkett@gmail.com>		2024-09-19 20:01:01 -06:00
..
models	Fixes to esrgan trainer. Moved logic for sd prompt embeddings out of diffusers pipeline so I can manipulate it	2023-09-16 17:41:07 -06:00
BaseExtensionProcess.py	Huge memory optimizations, many big fixes	2023-08-27 17:48:02 -06:00
BaseExtractProcess.py	Huge memory optimizations, many big fixes	2023-08-27 17:48:02 -06:00
BaseMergeProcess.py	Huge memory optimizations, many big fixes	2023-08-27 17:48:02 -06:00
BaseProcess.py	Massive speed increases and ram optimizations	2023-10-10 06:07:55 -06:00
BaseSDTrainProcess.py	Feat: Wandb logging (#95)	2024-09-19 20:01:01 -06:00
BaseTrainProcess.py	Bug fixes, speed improvements, compatability adjustments withdiffusers updates	2023-09-13 07:03:53 -06:00
ExtractLoconProcess.py	Adjusted some names and allow to ignore text encoder on extract	2023-07-16 12:52:39 -06:00
ExtractLoraProcess.py	Added support for 3cleir, not fully tested	2023-07-16 15:35:14 -06:00
GenerateProcess.py	New image generation img2img. various tweaks and fixes	2024-07-24 04:13:41 -06:00
MergeLoconProcess.py	Setup the base process for merging things. WIP	2023-07-20 07:39:31 -06:00
ModRescaleLoraProcess.py	Huge memory optimizations, many big fixes	2023-08-27 17:48:02 -06:00
TrainESRGANProcess.py	Added some further extendability for plugins	2023-09-19 05:41:44 -06:00
TrainFineTuneProcess.py	Setup base for training jobs. Added sd-scripts as a submodule	2023-07-08 13:50:59 -06:00
TrainSDRescaleProcess.py	Added inbuild plugins and made one for image referenced. WIP	2023-08-10 16:02:44 -06:00
TrainSliderProcess.py	Tons of bug fixes and improvements to special training. Fixed slider training.	2023-12-09 16:38:10 -07:00
TrainSliderProcessOld.py	Added inbuild plugins and made one for image referenced. WIP	2023-08-10 16:02:44 -06:00
TrainVAEProcess.py	Bug fixes and little improvements here and there.	2024-06-08 06:24:20 -06:00
__init__.py	Hude rework to move the batch to a DTO to make it far more modular to the future ui	2023-08-29 10:22:19 -06:00