ReconViaGen / requirements.txt
virtualbot123421's picture
Update requirements.txt
94c57d6 verified
--extra-index-url https://download.pytorch.org/whl/cu121
# Kept pinned to ensure compatibility with the flash_attn wheel below
torch==2.4.0
torchvision==0.19.0
xformers==0.0.27.post2
# Unpinned packages (versions removed for automatic resolution)
pillow
imageio
imageio-ffmpeg
tqdm
easydict
opencv-python-headless
scipy
rembg
onnxruntime
trimesh
xatlas
pyvista
pymeshfix
igraph
spconv-cu120
transformers
spaces
pydantic
# Direct repository/wheel links
git+https://github.com/EasternJournalist/utils3d.git@9a4eb15e4021b67b12c460c7057d642626897ec8
https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.0.post2/flash_attn-2.7.0.post2+cu12torch2.4cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
https://huggingface.co/spaces/JeffreyXiang/TRELLIS/resolve/main/wheels/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl?download=true
https://huggingface.co/spaces/JeffreyXiang/TRELLIS/resolve/main/wheels/nvdiffrast-0.3.3-cp310-cp310-linux_x86_64.whl?download=true