model_test_commit
cf004f4
verified
-
1.52 kB
initial commit
distilgpt2_squad_qa_model.bin
Detected Pickle imports (22)
- "transformers.models.gpt2.modeling_gpt2.GPT2Attention",
- "torch.nn.modules.dropout.Dropout",
- "torch.BoolStorage",
- "torch.nn.modules.container.ModuleList",
- "transformers.models.gpt2.modeling_gpt2.GPT2Block",
- "transformers.models.gpt2.configuration_gpt2.GPT2Config",
- "collections.OrderedDict",
- "transformers.models.gpt2.modeling_gpt2.GPT2LMHeadModel",
- "torch.nn.utils.prune.PruningContainer",
- "torch._utils._rebuild_parameter",
- "torch._utils._rebuild_tensor_v2",
- "transformers.pytorch_utils.Conv1D",
- "torch.nn.utils.prune.L1Unstructured",
- "transformers.activations.NewGELUActivation",
- "transformers.models.gpt2.modeling_gpt2.GPT2MLP",
- "torch.nn.modules.normalization.LayerNorm",
- "transformers.models.gpt2.modeling_gpt2.GPT2Model",
- "transformers.generation.configuration_utils.GenerationConfig",
- "torch.nn.modules.sparse.Embedding",
- "__builtin__.set",
- "torch.nn.modules.linear.Linear",
- "torch.FloatStorage"
How to fix it?
643 MB
model_test_commit