torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 transformers diffusers accelerate xformers opencv-python-headless pillow numpy matplotlib gradio==3.50.0 huggingface_hub psutil flash-attn==2.7.3