chigozienri / VQGAN-CLIP-animations

133 stars 28 forks source link

Several missing module errors in vqgan #40

Open jessesnider opened 1 year ago

jessesnider commented 1 year ago

Describe the bug Every time I try and download the image libraries, i get the set of errors below. This error consists throughout all VQGAN colab libraries I can find.

Parameters used errors happened before getting to parameters

Which cell you saw the error in Prepare Folders

Error message

ModuleNotFoundError Traceback (most recent call last) in <cell line: 50>() 48 from omegaconf import OmegaConf 49 from PIL import Image ---> 50 from taming.models import cond_transformer, vqgan 51 import torch 52 from torch import nn, optim

2 frames /content/./taming-transformers/taming/models/cond_transformer.py in 4 import pytorch_lightning as pl 5 ----> 6 from main import instantiate_from_config 7 from taming.modules.util import SOSProvider 8

/content/./taming-transformers/main.py in 12 from pytorch_lightning.utilities import rank_zero_only 13 ---> 14 from taming.data.utils import custom_collate 15 16

/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'


I was banking on having this tool available, but like I said, it seems all vqgan colabs I can find are relying on whatever is now broken.

MeanOgreGrr commented 1 year ago

I am a huge fan of this tool and am completely lost trying to figure out how to fix the issue described above by jessesnider. If someone in the know of how this operates could point us in the direction for how to get this back up and running, and possibly keep it up and running for the future, I know at least two folks who would be very thankful.

MeanOgreGrr commented 1 year ago

Found a solution: https://github.com/CompVis/taming-transformers/pull/202/commits/1b9e495e610d0b8ec371acda314a45964409898a