subjects-celebrity2024-05-16T17-22-46_zero3-ada-30000.pt
Detected Pickle imports (28)
- "torch.FloatStorage",
- "torch.nn.modules.container.ModuleDict",
- "transformers.models.clip.modeling_clip.CLIPTextTransformer",
- "adaface.arc2face_models.CLIPAttentionMKV",
- "adaface.subj_basis_generator.CrossAttention",
- "torch._utils._rebuild_tensor_v2",
- "adaface.arc2face_models.CLIPTextModelWrapper",
- "torch.nn.modules.linear.Identity",
- "torch.nn.modules.container.Sequential",
- "torch.nn.modules.normalization.LayerNorm",
- "collections.OrderedDict",
- "torch.BoolStorage",
- "torch.nn.modules.container.ModuleList",
- "transformers.models.clip.modeling_clip.CLIPTextEmbeddings",
- "torch.nn.modules.dropout.Dropout",
- "torch.nn.modules.linear.Linear",
- "transformers.models.clip.modeling_clip.CLIPMLP",
- "torch._utils._rebuild_parameter",
- "torch.LongStorage",
- "transformers.models.clip.modeling_clip.CLIPEncoder",
- "transformers.activations.QuickGELUActivation",
- "torch.nn.modules.sparse.Embedding",
- "__builtin__.set",
- "einops.layers.torch.Rearrange",
- "ldm.util.GradientScaler",
- "transformers.models.clip.configuration_clip.CLIPTextConfig",
- "adaface.subj_basis_generator.SubjBasisGenerator",
- "transformers.models.clip.modeling_clip.CLIPEncoderLayer"
How to fix it?
822 MB
re-init