pre-requirements.txt

--extra-index-url https://download.pytorch.org/whl/cu121
--extra-index-url https://data.dgl.ai/wheels/torch-2.1/cu121/repo.html
torch==2.2.0
torchvision==0.17.0
dgl
accelerate
beartype
einops
gateloop_transformer
matplotlib
scikit-learn
pandas
pytorch_custom_utils
gradio
pydantic==2.10.6
x_transformers
torch_redstone
torchdata==0.9.0
toolz
environs
jaxtyping
omegaconf
ema_pytorch
local_attention==1.9.15
taylor_series_linear_attention
transformers
vector_quantize_pytorch
open3d
trimesh
pytorch_lightning
scikit-image
opencv-python
mesh2sdf
seaborn
mesh_to_sdf
point_cloud_utils