diff --git a/cog.yaml b/cog.yaml
index bb0b8c5..4ac1b90 100644
--- a/cog.yaml
+++ b/cog.yaml
@@ -15,6 +15,7 @@ build:
     - "realesrgan==0.3.0"
     - "gdown==5.1.0"
     - "mediapipe==0.10.14"
+    - "setuptools==67.7.2"
   run:
     - curl -o /usr/local/bin/pget -L "https://github.com/replicate/pget/releases/latest/download/pget_$(uname -s)_$(uname -m)"
     - chmod +x /usr/local/bin/pget
diff --git a/requirements.txt b/requirements.txt
index 956fef2..0162bb9 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,7 +1,7 @@
 GitPython
 Pillow
 accelerate
-
+setuptools==67.7.2
 basicsr
 blendmodes
 clean-fid