torch torchvision tensorflow numpy transformers==4.45.2 peft protobuf sentencepiece pytest matplotlib scikit-learn scipy spacy einops jaxtyping diffusers==0.30.3 opencv-python git+https://github.com/facebookresearch/pytorch3d.git gradio==4.44.1