search

CompVis / taming-transformers

Taming Transformers for High-Resolution Image Synthesis
https://arxiv.org/abs/2012.09841
MIT License
5.83k stars 1.15k forks source link

ModuleNotFoundError: No module named 'torch._six' #223

Open sandrodevdariani opened 1 year ago

sandrodevdariani commented 1 year ago

guys it is deprecated.

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

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