diff --git a/.github/workflows/unit-test.yml b/.github/workflows/unit-test.yml
index 77c776c..8f5852a 100644
--- a/.github/workflows/unit-test.yml
+++ b/.github/workflows/unit-test.yml
@@ -23,7 +23,7 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        pip install -r requirements.txt
+        pip install -r requirements-test.txt
         
     - name: Run unit tests
       run: |
diff --git a/requirements-test.txt b/requirements-test.txt
new file mode 100644
index 0000000..5bb4e5c
--- /dev/null
+++ b/requirements-test.txt
@@ -0,0 +1,19 @@
+diffusers[torch]>=0.14.0
+ninja
+numpy
+omegaconf==2.2.3
+protobuf==3.20.3
+pyre-extensions==0.0.23
+pytorch-lightning==1.9.2
+transformers==4.27.1
+aiohttp==3.8.4
+bitsandbytes==0.37.2
+colorama==0.4.6
+compel~=1.1.3
+ftfy==6.1.1
+lion-pytorch
+piexif==1.1.3
+pyfakefs
+pynvml==11.5.0
+tensorboard==2.12.0
+wandb