TTI / Reward /robometer /requirements-robodopamine.txt
JosephBai's picture
Upload folder using huggingface_hub
857c2e9 verified
# Reward-fm dependencies for Robo-Dopamine venv.
# Install this file first: vLLM brings torch 2.9.x; we omit torch/torchvision/xformers here.
vllm
accelerate>=1.9.0
datasets==4.1.1
einops>=0.8.0
evaluate>=0.4.3
h5py>=3.10.0
huggingface-hub[hf-xet]>0.35
numpy>=1.21.0
matplotlib>=3.5.0
peft
pyrallis>=0.3.0
safetensors>=0.6.0rc0
sentence-transformers>=2.0.0
tqdm>=4.67.1
wandb>=0.20.1
pyyaml>=6.0.0
rich>=14.0.0
soundfile>=0.13.1
ipdb>=0.13.13
hf_transfer>=0.1.0
gradio>=4.0.0
pillow>=9.0.0
termcolor>=3.1.0
qwen-vl-utils[decord]==0.0.14
fastapi>=0.116.1
uvicorn>=0.35.0
requests>=2.32.4
google-generativeai
scipy>=1.15.3
codetiming>=1.4.0
robosuite==1.4.0
imageio>=2.37.0
omegaconf>=2.3.0
opencv-python-headless>=4.8.0
opencv-contrib-python-headless>=4.8.0
num2words
seaborn
bitsandbytes
moviepy
wandb[media]
torchcodec
decord>=0.6.0
unsloth>=2025.10
gymnasium>=1.2.2
metaworld>=3.0.0
loguru>=0.7.3
tensorboard
hydra-core
tensorflow-datasets==4.9.2
tensorflow==2.19
tf-keras>=2.19.0
protobuf<3.21