Add TVM and speedups
a07656a
-
1.48 kB
initial commit
GPT2Model.pt
Detected Pickle imports (19)
- "transformers.models.gpt2.modeling_gpt2.GPT2Block",
- "__builtin__.set",
- "torch.nn.modules.linear.Linear",
- "torch._utils._rebuild_parameter",
- "torch._utils._rebuild_tensor_v2",
- "torch.nn.modules.container.ModuleList",
- "torch.nn.modules.sparse.Embedding",
- "torch.ByteStorage",
- "transformers.models.gpt2.modeling_gpt2.GPT2MLP",
- "transformers.models.gpt2.modeling_gpt2.GPT2Model",
- "torch.nn.modules.normalization.LayerNorm",
- "transformers.models.gpt2.modeling_gpt2.GPT2Attention",
- "torch.nn.modules.dropout.Dropout",
- "collections.OrderedDict",
- "transformers.activations.NewGELUActivation",
- "transformers.models.gpt2.configuration_gpt2.GPT2Config",
- "transformers.pytorch_utils.Conv1D",
- "transformers.models.gpt2.modeling_gpt2.GPT2LMHeadModel",
- "torch.FloatStorage"
How to fix it?
510 MB
Upload GPT2Model.pt
GPT2Tokenizer.pt
Detected Pickle imports (5)
- "transformers.tokenization_utils.Trie",
- "regex._regex.compile",
- "transformers.models.gpt2.tokenization_gpt2.GPT2Tokenizer",
- "tokenizers.AddedToken",
- "_codecs.encode"
How to fix it?
3.1 MB
Upload 4 files
-
250 Bytes
initial commit
-
1.52 kB
Add TVM and speedups
-
156 Bytes
Update requirements.txt