ECON / requirements.txt
Yuliang's picture
update docker
2a5bdc2
raw
history blame
472 Bytes
--extra-index-url https://download.pytorch.org/whl/cu116
torch==1.13.1+cu116
torchvision==0.14.1+cu116
fvcore
iopath
pyembree
cupy-cuda11x
cython
matplotlib
scikit-image
trimesh
rtree
pytorch_lightning
kornia>0.4.0
chumpy
opencv-python-headless==4.8.1.78
opencv_contrib_python
scikit-learn
protobuf
dataclasses
mediapipe
einops
boto3
open3d
xatlas
transformers
controlnet_aux
xformers==0.0.16
triton
diffusers
accelerate
gradio
git+https://github.com/YuliangXiu/rembg.git