Upload checkpoint
3ebca22
-
1.38 kB
initial commit
-
21 Bytes
initial commit
checkpoint.pt
Detected Pickle imports (60)
- "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?
6.13 GB
Upload checkpoint