VQASynth / requirements.txt
salma-remyx's picture
update reqs
21ad39d
raw
history blame contribute delete
783 Bytes
--extra-index-url https://download.pytorch.org/whl/cu121
torch==2.4.0
torchvision==0.19.0
torchaudio==2.4.0
transformers==4.44.2
pillow==11.0.0
gradio==5.5.0
accelerate==0.34.2
numpy==1.26.4
timm==1.0.9
einops==0.7.0
open3d==0.18.0
opencv-python==4.7.0.72
tqdm==4.66.3
torchprofile==0.0.4
matplotlib==3.6.2
huggingface-hub==0.25.1
onnx==1.13.1
onnxruntime==1.14.1
onnxsim==0.4.35
scipy==1.12.0
litellm==1.25.2
pycocotools==2.0.6
datasets==3.1.0
spacy==3.7.5
pandas==2.2.3
html5lib==1.1
spaces==0.30.4
#git+https://github.com/remyxai/VQASynth.git
git+https://github.com/apple/ml-depth-pro.git
git+https://github.com/facebookresearch/sam2.git
git+https://github.com/openai/CLIP.git
flash-attn @ https://remyx.ai/assets/spatialvlm/flash_attn-2.7.0.post2-cp310-cp310-linux_x86_64.whl