Fetching metadata from the HF Docker repository...
Update main.py
c0734d1
verified
-
1.52 kB
initial commit
-
433 Bytes
Create Dockerfile
-
193 Bytes
initial commit
-
1.81 kB
Update main.py
model.pth
Detected Pickle imports (25)
- "transformers.models.roberta.modeling_roberta.RobertaEmbeddings",
- "torch.nn.modules.container.ModuleList",
- "transformers.models.roberta.modeling_roberta.RobertaOutput",
- "transformers.models.roberta.configuration_roberta.RobertaConfig",
- "transformers.models.roberta.modeling_roberta.RobertaForSequenceClassification",
- "collections.OrderedDict",
- "torch._C._nn.gelu",
- "transformers.activations.GELUActivation",
- "torch.nn.modules.normalization.LayerNorm",
- "transformers.models.roberta.modeling_roberta.RobertaSelfOutput",
- "torch._utils._rebuild_parameter",
- "transformers.models.roberta.modeling_roberta.RobertaIntermediate",
- "transformers.models.roberta.modeling_roberta.RobertaClassificationHead",
- "transformers.models.roberta.modeling_roberta.RobertaLayer",
- "torch.nn.modules.linear.Linear",
- "transformers.models.roberta.modeling_roberta.RobertaSdpaSelfAttention",
- "transformers.models.roberta.modeling_roberta.RobertaEncoder",
- "__builtin__.set",
- "torch._utils._rebuild_tensor_v2",
- "transformers.models.roberta.modeling_roberta.RobertaAttention",
- "torch.LongStorage",
- "transformers.models.roberta.modeling_roberta.RobertaModel",
- "torch.nn.modules.dropout.Dropout",
- "torch.FloatStorage",
- "torch.nn.modules.sparse.Embedding"
How to fix it?
499 MB
Upload model.pth with huggingface_hub
-
60 Bytes
Update requirements.txt