ECCV2022

/

storydalle

like 0

• "dalle.models.stage2.layers.MultiHeadSelfAttention",
• "dalle.models.stage1.vqgan.VQGAN",
• "torch.nn.modules.conv.Conv2d",
• "omegaconf.nodes.FloatNode",
• "dalle.utils.config.Stage2Hparams",
• "dalle.models.stage2.layers.GELU",
• "dalle.utils.config.StoryModelConfig",
• "dalle.models.stage1.layers.Decoder",
• "torch.nn.modules.container.ModuleList",
• "dalle.models.stage2.transformer.Transformer1d",
• "torch.nn.modules.normalization.LayerNorm",
• "dalle.utils.config.Stage1Config",
• "torch.nn.modules.dropout.Dropout",
• "torch.nn.modules.linear.Linear",
• "typing.List",
• "omegaconf.listconfig.ListConfig",
• "__builtin__.list",
• "dalle.models.stage1.layers.ResnetBlock",
• "torch._utils._rebuild_tensor_v2",
• "dalle.models.stage1.layers.Encoder",
• "omegaconf.nodes.BooleanNode",
• "dalle.models.stage1.vqgan.VectorQuantizer",
• "torch.float32",
• "__builtin__.set",
• "torch.device",
• "__builtin__.dict",
• "tokenizers.implementations.char_level_bpe.CharBPETokenizer",
• "_codecs.encode",
• "dalle.utils.config.DataConfig",
• "__builtin__.long",
• "collections.defaultdict",
• "omegaconf.nodes.StringNode",
• "dalle.utils.config.Stage1Hparams",
• "dalle.utils.config.StoryConfig",
• "dalle.models.stage1.layers.Upsample",
• "dalle.models.stage1.layers.AttnBlock",
• "dalle.utils.config.OptConfig",
• "__builtin__.unicode",
• "dalle.models.stage1.layers.Downsample",
• "torch.nn.modules.normalization.GroupNorm",
• "dalle.models.stage2.layers.Block",
• "omegaconf.base.ContainerMetadata",
• "tokenizers.models.Model",
• "omegaconf.base.Metadata",
• "collections.OrderedDict",
• "dalle.models.StoryDalle",
• "__builtin__.float",
• "dalle.utils.config.Stage2Config",
• "omegaconf.nodes.IntegerNode",
• "dalle.models.stage2.layers.CrossAttentionLayer",
• "tokenizers.Tokenizer",
• "torch._utils._rebuild_parameter",
• "torch.nn.modules.module.Module",
• "typing.Any",
• "__builtin__.bool",
• "torch.nn.modules.container.Sequential",
• "dalle.utils.config.ExpConfig",
• "torch.FloatStorage",
• "omegaconf.dictconfig.DictConfig",
• "torch.nn.modules.sparse.Embedding"

How to fix it?