Fetching metadata from the HF Docker repository...
Update requirements.txt
25ef1d0
-
1.48 kB
initial commit
-
239 Bytes
initial commit
-
902 Bytes
Update app.py
poem_model.pt
Detected Pickle imports (20)
- "transformers.generation.configuration_utils.GenerationConfig",
- "torch.FloatStorage",
- "torch.nn.modules.linear.Linear",
- "transformers.models.gpt2.modeling_gpt2.GPT2Model",
- "__builtin__.set",
- "torch._utils._rebuild_parameter",
- "torch.nn.modules.sparse.Embedding",
- "transformers.models.gpt2.configuration_gpt2.GPT2Config",
- "torch._utils._rebuild_tensor_v2",
- "transformers.models.gpt2.modeling_gpt2.GPT2Attention",
- "transformers.activations.NewGELUActivation",
- "torch.nn.modules.normalization.LayerNorm",
- "torch.ByteStorage",
- "torch.nn.modules.dropout.Dropout",
- "transformers.pytorch_utils.Conv1D",
- "transformers.models.gpt2.modeling_gpt2.GPT2MLP",
- "collections.OrderedDict",
- "transformers.models.gpt2.modeling_gpt2.GPT2Block",
- "transformers.models.gpt2.modeling_gpt2.GPT2LMHeadModel",
- "torch.nn.modules.container.ModuleList"
How to fix it?
510 MB
Upload poem_model.pt
-
310 kB
Upload poems_data.json
-
19 Bytes
Update requirements.txt