Spaces:

RaviNaik
/

ERA-SESSION21

Runtime error

App Files Files Community

ERA-SESSION21 / checkpoints

52.7 MB

Ctrl+K

Ctrl+K

2 contributors

History: 1 commit

ravi.naik

Added training, inference and gradio UI code

213d16f almost 3 years ago

model.pth
Detected Pickle imports (17)
- "src.model.Decoder",
- "torch.nn.modules.container.ModuleList",
- "src.model.MultiHeadAttention",
- "src.model.Head",
- "torch.nn.modules.normalization.LayerNorm",
- "collections.OrderedDict",
- "src.model.GPTModel",
- "torch._utils._rebuild_tensor_v2",
- "torch.nn.modules.container.Sequential",
- "torch.nn.modules.dropout.Dropout",
- "torch.nn.modules.linear.Linear",
- "__builtin__.set",
- "torch.FloatStorage",
- "torch._utils._rebuild_parameter",
- "src.model.FeedForward",
- "torch.nn.modules.sparse.Embedding",
- "torch.nn.modules.activation.ReLU"
How to fix it?
52.7 MB
xet

Added training, inference and gradio UI code almost 3 years ago