transformers==5.2.0 accelerate safetensors draccus einops diffusers av datasets pandas pyarrow numpy pillow matplotlib torchvision imageio[ffmpeg] mediapy jsonlines omegaconf termcolor deepdiff opencv-python-headless scipy