Update inference.py
d09fda3
-
1.48 kB
initial commit
-
374 Bytes
Update README.md
-
3.49 kB
Update inference.py
passage_model_3.pt
Detected Pickle imports (26)
- "transformers.models.bert.modeling_bert.BertIntermediate",
- "torch.LongStorage",
- "transformers.models.bert.modeling_bert.BertAttention",
- "torch.nn.modules.container.ModuleList",
- "torch.nn.modules.activation.Tanh",
- "__main__.BertSimilarity",
- "transformers.models.bert.modeling_bert.BertPooler",
- "torch.nn.modules.sparse.Embedding",
- "torch.nn.modules.normalization.LayerNorm",
- "transformers.models.bert.modeling_bert.BertOutput",
- "torch.nn.modules.linear.Linear",
- "torch.nn.modules.dropout.Dropout",
- "transformers.models.bert.configuration_bert.BertConfig",
- "torch._utils._rebuild_tensor_v2",
- "collections.OrderedDict",
- "__builtin__.set",
- "transformers.models.bert.modeling_bert.BertSelfAttention",
- "torch._utils._rebuild_parameter",
- "torch.FloatStorage",
- "transformers.models.bert.modeling_bert.BertSelfOutput",
- "transformers.activations.GELUActivation",
- "transformers.models.bert.modeling_bert.BertModel",
- "transformers.models.bert.modeling_bert.BertEmbeddings",
- "transformers.models.bert.modeling_bert.BertEncoder",
- "torch._C._nn.gelu",
- "transformers.models.bert.modeling_bert.BertLayer"
How to fix it?
438 MB
Added model weights
query_model_3.pt
Detected Pickle imports (26)
- "torch._utils._rebuild_parameter",
- "__main__.BertSimilarity",
- "torch.nn.modules.dropout.Dropout",
- "torch.nn.modules.normalization.LayerNorm",
- "torch.nn.modules.container.ModuleList",
- "transformers.activations.GELUActivation",
- "transformers.models.bert.modeling_bert.BertSelfOutput",
- "torch.LongStorage",
- "transformers.models.bert.modeling_bert.BertIntermediate",
- "torch._utils._rebuild_tensor_v2",
- "torch.nn.modules.linear.Linear",
- "transformers.models.bert.modeling_bert.BertPooler",
- "torch.FloatStorage",
- "torch.nn.modules.sparse.Embedding",
- "torch.nn.modules.activation.Tanh",
- "transformers.models.bert.modeling_bert.BertModel",
- "transformers.models.bert.modeling_bert.BertEncoder",
- "collections.OrderedDict",
- "transformers.models.bert.configuration_bert.BertConfig",
- "transformers.models.bert.modeling_bert.BertAttention",
- "transformers.models.bert.modeling_bert.BertEmbeddings",
- "torch._C._nn.gelu",
- "transformers.models.bert.modeling_bert.BertSelfAttention",
- "__builtin__.set",
- "transformers.models.bert.modeling_bert.BertOutput",
- "transformers.models.bert.modeling_bert.BertLayer"
How to fix it?
438 MB
Added model weights
