Fetching metadata from the HF Docker repository...
Upload best_model.pkl
27dad6b
verified
-
1.52 kB
initial commit
-
255 Bytes
initial commit
best_model.pkl
Detected Pickle imports (26)
- "transformers.models.gpt2.modeling_gpt2.GPT2Attention",
- "torch.nn.modules.normalization.LayerNorm",
- "__builtin__.set",
- "transformers.models.gpt2.modeling_gpt2.GPT2MLP",
- "torch.BoolStorage",
- "transformers.activations.NewGELUActivation",
- "collections.OrderedDict",
- "transformers.models.gpt2.configuration_gpt2.GPT2Config",
- "torch.nn.modules.container.ModuleList",
- "transformers.generation.configuration_utils.GenerationConfig",
- "tokenizers.AddedToken",
- "torch.nn.modules.linear.Linear",
- "transformers.models.gpt2.modeling_gpt2.GPT2LMHeadModel",
- "transformers.models.gpt2.modeling_gpt2.GPT2Block",
- "regex._regex.compile",
- "transformers.models.gpt2.modeling_gpt2.GPT2Model",
- "__builtin__.getattr",
- "torch.FloatStorage",
- "transformers.tokenization_utils.Trie",
- "torch.nn.modules.dropout.Dropout",
- "transformers.pytorch_utils.Conv1D",
- "_codecs.encode",
- "torch._utils._rebuild_parameter",
- "torch.nn.modules.sparse.Embedding",
- "torch._utils._rebuild_tensor_v2",
- "transformers.models.gpt2.tokenization_gpt2.GPT2Tokenizer"
How to fix it?
514 MB
Upload best_model.pkl