VQASynth / requirements.txt
salma-remyx's picture
fix torch version
b23d632
raw
history blame
719 Bytes
--extra-index-url https://download.pytorch.org/whl/cu121
torch==2.4.0
torchvision==0.19.0
torchaudio==2.4.0
transformers>=4.41.0
Pillow
gradio
accelerate==0.34.2
numpy==1.26.3
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.24.7
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
onnxruntime-gpu
pandas
html5lib
spaces
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