lightning==2.0.7 pytorch-lightning==2.0.2 plyfile OmegaConf matplotlib einops gradio diffusers==0.19.3 transformers==4.34.1 rembg segment_anything jaxtyping imageio imageio-ffmpeg