Update deps

YanWenKun · Jun 30, 2024 · 2384a82 · 2384a82
1 parent deed9dd
commit 2384a82
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -32,7 +32,7 @@ nerfacc
 ninja
 numba
 numexpr
-numpy
+numpy<2
 omegaconf
 onnxruntime
 onnxruntime-gpu
@@ -44,7 +44,7 @@ packaging
 pandas
 piexif
 pilgram
-pillow>=10.2.0
+pillow
 plyfile
 psutil
 py-cpuinfo
@@ -57,6 +57,7 @@ pymeshlab
 pynvml
 python-dateutil
 python-ffmpeg
+pytorch-lightning
 pytorch-msssim
 pyyaml
 qrcode[pil]
@@ -72,9 +73,12 @@ scikit-learn
 scipy
 segment-anything
 simpleeval
+soundfile
+spandrel
 svglib
 timm
 torch
+torchaudio
 torchdiffeq
 torchmetrics
 torchsde
@@ -83,6 +87,8 @@ torchvision
 tqdm
 transformers
 trimesh[easy]
+typer
+typing-extensions
 ultralytics!=8.2.5
 xatlas
 yacs

diff --git a/stage1.sh b/stage1.sh
@@ -34,7 +34,7 @@ curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py
 
 # PIP install
 ./python.exe -s -m pip install \
-    --upgrade pip wheel setuptools Cython numpy cmake
+    --upgrade pip wheel setuptools Cython cmake
 
 ./python.exe -s -m pip install \
     xformers==0.0.25.post1 torchvision torchaudio \