tqdm scikit-image gdown torch torchvision gdown omegaconf tensorflow-hub pytorch-lightning einops tokenizers tensorflow allennlp==2.10.0