Fetching metadata from the HF Docker repository... 
minor changes
6fa365b - 1.17 kB initial commit
Gpt_neo_Epoch_10_Loss_031_data_5000.pth Detected Pickle imports (19)
- "torch._utils._rebuild_tensor_v2",
- "collections.OrderedDict",
- "torch._utils._rebuild_parameter",
- "transformers.models.gpt_neo.modeling_gpt_neo.GPTNeoBlock",
- "transformers.models.gpt_neo.modeling_gpt_neo.GPTNeoAttention",
- "torch.nn.modules.sparse.Embedding",
- "torch.nn.modules.normalization.LayerNorm",
- "torch.nn.modules.dropout.Dropout",
- "torch.nn.modules.linear.Linear",
- "torch.ByteStorage",
- "transformers.models.gpt_neo.modeling_gpt_neo.GPTNeoModel",
- "transformers.models.gpt_neo.modeling_gpt_neo.GPTNeoSelfAttention",
- "transformers.activations.NewGELUActivation",
- "__builtin__.set",
- "torch.nn.modules.container.ModuleList",
- "transformers.models.gpt_neo.configuration_gpt_neo.GPTNeoConfig",
- "transformers.models.gpt_neo.modeling_gpt_neo.GPTNeoMLP",
- "torch.FloatStorage",
- "transformers.models.gpt_neo.modeling_gpt_neo.GPTNeoForCausalLM"
How to fix it?
551 MB added model - 253 Bytes initial commit
- 2.56 kB minor changes
- 28 Bytes added model
- 10.8 kB added model
