pytorch_lightning==2.3
deepspeed taming-transformers scipy einops kornia open_clip_torch openai-clip xformers timm av gradio
