WorldMem_Repro / requirements.txt
BonanDing's picture
Reproduce Training & Fix distributed eval
681f346
aiofiles==23.2.1
aiohappyeyeballs==2.6.1
aiohttp==3.13.3
aiosignal==1.4.0
altair==5.5.0
annotated-doc==0.0.4
antlr4-python3-runtime==4.9.3
anyio==4.12.1
async-timeout==5.0.1
attrs==25.4.0
av==16.1.0
certifi==2026.1.4
charset-normalizer==3.4.4
click==8.3.1
colorama==0.4.6
colorlog==6.10.1
contourpy==1.3.2
cycler==0.12.1
decorator==4.4.2
diffusers==0.36.0
docker-pycreds==0.4.0
einops==0.8.1
exceptiongroup==1.3.1
fastapi==0.125.0
ffmpy==1.0.0
filelock==3.20.3
fonttools==4.61.1
frozenlist==1.8.0
fsspec==2024.12.0
fvcore==0.1.5.post20221221
gitdb==4.0.12
GitPython==3.1.46
gluonts==0.13.1
gradio==3.50.2
gradio_client==0.6.1
h11==0.16.0
h5py==3.15.1
hf-xet==1.2.0
httpcore==1.0.9
httpx==0.28.1
huggingface_hub==1.3.2
hydra-core==1.3.2
idna==3.11
ImageIO==2.37.2
imageio-ffmpeg==0.6.0
importlib_metadata==8.7.1
importlib_resources==6.5.2
internetarchive==5.7.1
iopath==0.1.10
Jinja2==3.1.6
jsonpatch==1.33
jsonpointer==3.0.0
jsonschema==4.26.0
jsonschema-specifications==2025.9.1
kiwisolver==1.4.9
lightning==2.1.4
lightning-utilities==0.15.2
lpips==0.1.4
MarkupSafe==2.1.5
matplotlib==3.10.8
moviepy==1.0.3
mpmath==1.3.0
multidict==6.7.0
narwhals==2.15.0
networkx==3.4.2
numpy==1.26.4
nvidia-cublas-cu12==12.1.3.1
nvidia-cuda-cupti-cu12==12.1.105
nvidia-cuda-nvrtc-cu12==12.1.105
nvidia-cuda-runtime-cu12==12.1.105
nvidia-cudnn-cu12==9.1.0.70
nvidia-cufft-cu12==11.0.2.54
nvidia-curand-cu12==10.3.2.106
nvidia-cusolver-cu12==11.4.5.107
nvidia-cusparse-cu12==12.1.0.106
nvidia-nccl-cu12==2.20.5
nvidia-nvjitlink-cu12==12.9.86
nvidia-nvtx-cu12==12.1.105
omegaconf==2.3.0
opencv-python==4.11.0.86
orjson==3.11.5
packaging==24.2
pandas==2.3.3
parameterized==0.9.0
pillow==10.4.0
platformdirs==4.5.1
portalocker==3.2.0
proglog==0.1.12
propcache==0.4.1
protobuf==3.19.6
psutil==5.9.8
pydantic==1.10.26
pydub==0.25.1
pyparsing==3.3.1
pyrealsense2==2.56.5.9235
python-dateutil==2.9.0.post0
python-multipart==0.0.21
pytorch-lightning==2.6.0
pytorchvideo==0.1.5
pytz==2025.2
PyYAML==6.0.3
pyzmq==27.1.0
referencing==0.37.0
regex==2026.1.15
requests==2.32.5
rotary-embedding-torch==0.8.9
rpds-py==0.30.0
safetensors==0.7.0
scipy==1.15.3
semantic-version==2.10.0
sentry-sdk==2.49.0
setproctitle==1.3.7
shellingham==1.5.4
six==1.17.0
smmap==5.0.2
spaces==0.46.0
starlette==0.50.0
sympy==1.14.0
tabulate==0.9.0
termcolor==3.3.0
timm==1.0.24
toolz==0.12.1
torch==2.4.1
torch-fidelity==0.3.0
torchmetrics==0.11.4
torchvision==0.19.1
tqdm==4.67.1
triton==3.0.0
typer-slim==0.21.1
typing_extensions==4.15.0
tzdata==2025.3
urllib3==2.6.3
uvicorn==0.40.0
wandb==0.17.9
wandb_osh==1.2.1
websockets==11.0.3
yacs==0.1.8
yarl==1.22.0
zipp==3.23.0