torch torchvision pytorch-lightning numpy tqdm einops omegaconf diffusers==0.15.0 transformers opencv-python huggingface-hub==0.25.0