Merge pull request #151 from OPHoperHPO/reqs-update

updated deps

Dockerfile.cuda

@@ -14,7 +14,7 @@ COPY ./carvekit/ml/files/models_loc.py ./carvekit/ml/files/models_loc.py
 RUN python3 -c "from carvekit.ml.files.models_loc import download_all; download_all();"
 RUN rm -rf ./carvekit
 
-FROM pytorch/pytorch:2.1.2-cuda12.1-cudnn8-runtime
+FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime
 WORKDIR /app
 
 RUN apt-get update && apt-get -y install libgl1 libglib2.0-0  # Install cv2 dep.

carvekit/__init__.py

@@ -1 +1 @@
-version = "4.1.1"
+version = "4.1.2"

requirements.txt

@@ -1,16 +1,16 @@
-requests~=2.31.0
-torch~=2.1.2
-Pillow==10.1.0
-typing~=3.7.4.3
-torchvision~=0.16.2
-opencv-python~=4.8.1.78
-numpy~=1.26.2
-loguru~=0.7.2
-uvicorn~=0.25.0
-fastapi~=0.108.0
-pydantic~=2.5.3
-click~=8.1.7
-tqdm~=4.66.1
-setuptools~=69.0.3
-aiofiles~=23.2.1
-python-multipart~=0.0.6
+requests>=2.31.0
+torch>=2.2.2
+Pillow>=10.3.0
+typing>=3.7.4.3
+torchvision>=0.17.2
+opencv-python>=4.9.0.80
+numpy>=1.26.4
+loguru>=0.7.2
+uvicorn>=0.29.0
+fastapi>=0.110.1
+pydantic>=2.6.4
+click>=8.1.7
+tqdm>=4.66.2
+setuptools>=69.2.0
+aiofiles>=23.2.1
+python-multipart>=0.0.9

requirements_dev.txt

@@ -1,2 +1,2 @@
-pre-commit==3.6.0
+pre-commit>=3.7.0
 

requirements_test.txt

@@ -1,2 +1,2 @@
-pytest~=7.4.3
+pytest>=8.1.1
 

setup.py

@@ -25,7 +25,7 @@ def req_file(filename: str, folder: str = "."):
     # you may also want to remove whitespace characters
     # Example: `\n` at the end of each line
     if os.getenv("COLAB_PACKAGE_RELEASE") is not None:
-        return [re.sub("(~=.*)|(==.*)|(typing.*)", "", x.strip()) for x in content]
+        return [re.sub("(>=.*)|(~=.*)|(==.*)|(typing.*)", "", x.strip()) for x in content]
     return [x.strip() for x in content]