d-robotics-rdt/RDT-170M/requirements.txt
2025-10-26 16:23:28 +08:00

25 lines
439 B
Plaintext

numpy<2.0
packaging==24.0
wandb==0.17.0
deepspeed==0.14.2
accelerate==0.30.1
diffusers==0.27.2
timm==1.0.3
transformers==4.41.0
sentencepiece==0.2.0
h5py==3.11.0
opencv-python==4.9.0.80
imgaug==0.4.0
pytz==2022.1
huggingface_hub==0.23.0
# requirements_data.txt
# tfds-nightly==4.9.4.dev202402070044
gsutil==5.27
tensorflow==2.15.0.post1
pillow==10.2.0
pyyaml==6.0.1
tensorflow-graphics==2021.12.3
imageio==2.34.0
imageio-ffmpeg==0.4.9