diff --git a/requirements.txt b/requirements.txt
index fd694c2..a8f72ec 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -18,4 +18,5 @@ lion-pytorch
 compel~=1.1.3
 OmegaConf==2.2.3
 numpy==1.23.5
-wandb
\ No newline at end of file
+wandb
+colorama