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

__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

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