adapted for newer torch added pip requirements.txt

Author: yGifoom

requirements.txt

@@ -0,0 +1,34 @@
+pip>=20.3
+torch>=1.7.0
+torchvision>=0.8.1
+numpy>=1.19.2
+albumentations>=0.4.3
+opencv-python>=4.1.2.30
+pudb>=2019.2
+imageio>=2.9.0
+imageio-ffmpeg>=0.4.2
+pytorch-lightning>=1.0.8
+omegaconf>=2.0.0
+test-tube>=0.7.5
+streamlit>=0.73.1
+einops>=0.3.0
+more-itertools>=8.0.0
+transformers>=4.3.1
+-epython>=3.8.5
+pip>=20.3
+cudatoolkit>=10.2
+torch>=1.7.0
+torchvision>=0.8.1
+numpy>=1.19.2
+albumentations>=0.4.3
+opencv-python>=4.1.2.30
+pudb>=2019.2
+imageio>=2.9.0
+imageio-ffmpeg>=0.4.2
+pytorch-lightning>=1.0.8
+omegaconf>=2.0.0
+test-tube>=0.7.5
+streamlit>=0.73.1
+einops>=0.3.0
+more-itertools>=8.0.0
+transformers>=4.3.1

taming/data/utils.py

@@ -8,7 +8,7 @@
 import numpy as np
 import torch
 from taming.data.helper_types import Annotation
-from torch._six import string_classes
+string_classes = str
 from torch.utils.data._utils.collate import np_str_obj_array_pattern, default_collate_err_msg_format
 from tqdm import tqdm