Fetching metadata from the HF Docker repository... 
Upload best_model.pt
f613bf1 verified - 1.57 kB Upload 5 files
- 679 Bytes Update Dockerfile
- 195 Bytes initial commit
- 163 kB Upload best_model.pt
- 7.53 kB Update main.py
- 41.7 MB Upload 5 files
- 135 Bytes Update requirements.txt
traffic_transformer_model_2.pth Detected Pickle imports (17)
- "torch.nn.modules.activation.MultiheadAttention",
- "torch.nn.modules.linear.Linear",
- "torch._utils._rebuild_tensor_v2",
- "torch._C._nn.gelu",
- "torch.nn.modules.activation.ReLU",
- "torch.nn.modules.container.ModuleList",
- "torch.nn.modules.transformer.TransformerEncoderLayer",
- "collections.OrderedDict",
- "__main__.TransformerForecaster",
- "torch.FloatStorage",
- "torch.nn.modules.transformer.TransformerEncoder",
- "torch.nn.modules.container.Sequential",
- "torch.nn.modules.linear.NonDynamicallyQuantizableLinear",
- "torch.nn.modules.dropout.Dropout",
- "__builtin__.set",
- "torch.nn.modules.normalization.LayerNorm",
- "torch._utils._rebuild_parameter"
How to fix it?
172 kB Upload 5 files
