search

EleutherAI / vqgan-clip

MIT License
349 stars 40 forks source link

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

Open sandrodevdariani opened 1 year ago

sandrodevdariani commented 1 year ago

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 commented 1 year ago

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

it works for me.