ModuleNotFoundError in VQGAN+CLIP (codebook sampling method).ipynb

[sandrodevdariani]

4th cell

---

ModuleNotFoundError Traceback (most recent call last) in <cell line: 13>() 11 from PIL import Image 12 import requests ---> 13 from taming.models import cond_transformer, vqgan 14 import torch 15 from torch import nn, optim

2 frames /content/taming-transformers/taming/data/utils.py in 9 import torch 10 from taming.data.helper_types import Annotation ---> 11 from torch._six import string_classes 12 from torch.utils.data._utils.collate import np_str_obj_array_pattern, default_collate_err_msg_format 13 from tqdm import tqdm

ModuleNotFoundError: No module named 'torch._six'

[sandrodevdariani]

instead of from torch._six import string_classes. use string_classes = str

it works for me.