Fetching metadata from the HF Docker repository...
Upload nmt_model.pkl
44acb21
verified
-
1.52 kB
initial commit
-
311 Bytes
initial commit
-
960 Bytes
Create app.py
nmt_model.pkl
Detected Pickle imports (20)
- "transformers.models.marian.modeling_marian.MarianMTModel",
- "transformers.models.marian.modeling_marian.MarianAttention",
- "transformers.models.marian.modeling_marian.MarianDecoderLayer",
- "torch.nn.modules.normalization.LayerNorm",
- "torch.storage._load_from_bytes",
- "torch.nn.modules.sparse.Embedding",
- "torch.nn.modules.activation.SiLU",
- "transformers.models.marian.modeling_marian.MarianEncoder",
- "transformers.models.marian.modeling_marian.MarianModel",
- "torch._utils._rebuild_tensor_v2",
- "transformers.generation.configuration_utils.CompileConfig",
- "transformers.models.marian.configuration_marian.MarianConfig",
- "transformers.models.marian.modeling_marian.MarianDecoder",
- "transformers.generation.configuration_utils.GenerationConfig",
- "torch.nn.modules.linear.Linear",
- "torch._utils._rebuild_parameter",
- "collections.OrderedDict",
- "transformers.models.marian.modeling_marian.MarianEncoderLayer",
- "transformers.models.marian.modeling_marian.MarianSinusoidalPositionalEmbedding",
- "torch.nn.modules.container.ModuleList"
How to fix it?
308 MB
Upload nmt_model.pkl
-
81 Bytes
Create requirements.txt
-
2.61 kB
Create train.py