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