Refactoring the SD 2.x and SDXL trainers to be more in-line: missing …

…imports
bghira · Sep 14, 2023 · d3a1a0f · d3a1a0f
1 parent bdf849c
commit d3a1a0f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/helpers/legacy/validation.py b/helpers/legacy/validation.py
@@ -1,5 +1,5 @@
 import logging, torch, numpy as np
-
+from tqdm import tqdm
 from diffusers.utils import is_wandb_available
 from helpers.image_manipulation.brightness import calculate_luminance
 from diffusers import (