vmem / requirements.txt
liguang0115's picture
Update requirements.txt to use GitHub repository for curpo model dependency
7f6c8e0
raw
history blame
559 Bytes
--extra-index-url https://download.pytorch.org/whl/nightly/cu124
torch==2.5.1
torchvision==0.20.1
pydantic
gradio
matplotlib
tqdm
opencv-python
scipy
einops
trimesh
tensorboard
transformers
pyglet<2
huggingface-hub[torch]
pillow-heif
pyrender
kapture
kapture-localization
numpy==1.24.4
numpy-quaternion
pycolmap # for pnp
poselib # for pnp
viser
tyro
ninja
colorama
pytorch-lightning
splines
diffusers
kornia
open-clip-torch
accelerate
imageio[ffmpeg]
roma
spaces
omegaconf
wandb
evo
open3d
-e git+https://github.com/runjiali-rl/curope.git@main#egg=curope