Added training for a custom version of ERSGAN arcitecture. Testing training now

2026-05-01 03:31:35 +00:00 · 2023-08-07 18:04:23 -06:00
parent 8c90fa86c6
commit 8bd536df7e
9 changed files with 1694 additions and 141 deletions
--- a/jobs/TrainJob.py
+++ b/jobs/TrainJob.py
@@ -20,6 +20,7 @@ process_dict = {
    'slider_old': 'TrainSliderProcessOld',
    'lora_hack': 'TrainLoRAHack',
    'rescale_sd': 'TrainSDRescaleProcess',
+    'esrgan': 'TrainESRGANProcess',
 }