diff --git a/requirements.txt b/requirements.txt
index 8698f1bc..02702293 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,3 +4,4 @@ matplotlib
 opencv-python
 torch
 torchvision
+xformers