Fetching metadata from the HF Docker repository... 
dependencies
4f164af verified - 1.52 kB initial commit
- 307 Bytes initial commit
- 10.4 kB upload main file
- 160 Bytes dependencies
- 137 Bytes dependencies
review_classifier_model.pth Detected Pickle imports (15)
- "__builtin__.set",
- "__main__.MultiHeadAttention",
- "__main__.FeedForward",
- "torch._utils._rebuild_tensor_v2",
- "__main__.GELU",
- "torch.nn.modules.dropout.Dropout",
- "__main__.TransformerBlock",
- "torch.nn.modules.linear.Linear",
- "torch.nn.modules.container.Sequential",
- "torch._utils._rebuild_parameter",
- "collections.OrderedDict",
- "torch.FloatStorage",
- "__main__.GPTModel",
- "__main__.LayerNorm",
- "torch.nn.modules.sparse.Embedding"
How to fix it?
548 MB architecture model - 52 Bytes tokenizer file
- 134 Bytes dependencies
