diff --git a/helpers/legacy/validation.py b/helpers/legacy/validation.py
index 12bc45d5..f57c7b31 100644
--- 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 (