requirements.txt

--extra-index-url https://download.pytorch.org/whl/cu117
torch == 2.0.1
torchvision == 0.15.2
xformers == 0.0.21
diffusers == 0.25.0
opencv-python
numpy == 1.23.1
Pillow
h5py
matplotlib==3.7.5
scikit-image
gradio==4.43.0
gradio-imageslider==0.0.16
trimesh==4.0.5
tqdm == 4.65.0
accelerate==0.28.0
transformers==4.39.1
imgaug
GitPython==3.1.40
pygltflib==1.16.1
fire
datetime
rembg
segment_anything
pyvista
cupy-cuda11x
trimesh