Validations: fix

bghira · Sep 15, 2023 · d68dfa4 · d68dfa4
1 parent f2108e3
commit d68dfa4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/train_sd21.py b/train_sd21.py
@@ -998,6 +998,7 @@ def collate_fn(batch):
                 unet,
                 args,
                 validation_prompts,
+                validation_shortnames,
                 global_step,
                 resume_global_step,
                 step,

diff --git a/train_sdxl.py b/train_sdxl.py
@@ -1177,6 +1177,7 @@ def collate_fn(batch):
                 unet,
                 args,
                 validation_prompts,
+                validation_shortnames,
                 global_step,
                 resume_global_step,
                 step,