addict datasets modelscope pyarrow==20.0.0 xformers torch==2.1.2 open_clip_torch>=2.0.2 opencv-python-headless opencv-python einops>=0.4 rotary-embedding-torch fairscale scipy imageio pytorch-lightning torchsde easydict spaces bitsandbytes numpy<2