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.