torch flash-attn transformers peft protobuf sentencepiece pytest matplotlib scikit-learn scipy spacy einops jaxtyping diffusers opencv-python==4.5.5.64 opencv-python-headless==4.5.5.64 git+https://github.com/facebookresearch/pytorch3d.git