diff --git a/requirements-validator.txt b/requirements-validator.txt
index e7bb3f6..6a51f99 100644
--- a/requirements-validator.txt
+++ b/requirements-validator.txt
@@ -1,4 +1,3 @@
 httpx==0.27.0
 diffusers==0.30.0
 transformers==4.44.0
-wandb==0.17.6
diff --git a/requirements.txt b/requirements.txt
index fe22eea..ab0eeb3 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -16,3 +16,4 @@ datasets==2.20.0
 accelerate==0.33.0
 tensorboardx==2.6.2.2
 tf-keras==2.17.0
+wandb==0.17.6