adaface-neurips
fix style switching bugs
d2b3308
VGGface2_HQ_masks2024-10-14T16-09-24_zero3-ada-3500.pt
Detected Pickle imports (46)
- "adaface.subj_basis_generator.CrossAttention",
- "torch.nn.modules.container.ModuleDict",
- "__builtin__.dict",
- "transformers.models.clip.modeling_clip.CLIPEncoderLayer",
- "torch.nn.modules.container.Sequential",
- "einops.layers.torch.Rearrange",
- "omegaconf.listconfig.ListConfig",
- "transformers.models.clip.modeling_clip.CLIPTextTransformer",
- "transformers.models.clip.configuration_clip.CLIPTextConfig",
- "torch.nn.modules.linear.Identity",
- "transformers.activations.QuickGELUActivation",
- "collections.defaultdict",
- "omegaconf.base.Metadata",
- "omegaconf.base.ContainerMetadata",
- "adaface.subj_basis_generator.SubjBasisGenerator",
- "adaface.arc2face_models.CLIPTextModelWrapper",
- "torch.float32",
- "_codecs.encode",
- "typing.Any",
- "omegaconf.dictconfig.DictConfig",
- "torch.FloatStorage",
- "adaface.util.GradientScaler",
- "tokenizers.AddedToken",
- "transformers.models.clip.modeling_clip.CLIPEncoder",
- "transformers.models.clip.tokenization_clip.CLIPTokenizer",
- "torch.LongStorage",
- "__builtin__.list",
- "adaface.arc2face_models.CLIPAttentionMKV",
- "omegaconf.nodes.AnyNode",
- "transformers.models.clip.modeling_clip.CLIPTextEmbeddings",
- "__builtin__.set",
- "torch._utils._rebuild_tensor_v2",
- "__builtin__.long",
- "torch.nn.modules.linear.Linear",
- "transformers.tokenization_utils.Trie",
- "transformers.models.clip.modeling_clip.CLIPMLP",
- "ftfy.fix_text",
- "torch.BoolStorage",
- "torch.nn.modules.container.ModuleList",
- "regex._regex.compile",
- "torch.nn.modules.dropout.Dropout",
- "torch.nn.modules.batchnorm.BatchNorm2d",
- "collections.OrderedDict",
- "torch.nn.modules.normalization.LayerNorm",
- "torch.nn.modules.sparse.Embedding",
- "torch._utils._rebuild_parameter"
How to fix it?
1.81 GB
fix style switching bugs