diff --git a/requirements.txt b/requirements.txt
index b8de11d..f995f54 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -20,6 +20,7 @@ torch
 torchvision
 kornia
 opencv-contrib-python
+wandb
 
 # RAG dependencies
 langchain>=0.1.0