torch==1.12.1 torchvision==0.13.1 accelerate==0.12.0 absl-py ml_collections einops ftfy==6.1.1 transformers==4.23.1 loguru gradio==3.34.0 omegaconf wget xformers=0.0.16