MRNH

/

mbart-italian-grammar-corrector

like 1

• "__builtin__.set",
• "transformers.models.mbart.modeling_mbart.MBartForConditionalGeneration",
• "torch.nn.modules.normalization.LayerNorm",
• "torch.nn.modules.sparse.Embedding",
• "transformers.models.mbart.modeling_mbart.MBartLearnedPositionalEmbedding",
• "transformers.models.mbart.modeling_mbart.MBartDecoder",
• "transformers.models.mbart.modeling_mbart.MBartModel",
• "torch.FloatStorage",
• "torch._C._nn.gelu",
• "transformers.generation.configuration_utils.GenerationConfig",
• "transformers.models.mbart.modeling_mbart.MBartDecoderLayer",
• "transformers.models.mbart.modeling_mbart.MBartAttention",
• "collections.OrderedDict",
• "transformers.activations.GELUActivation",
• "torch._utils._rebuild_parameter",
• "transformers.models.mbart.modeling_mbart.MBartEncoder",
• "torch.nn.modules.linear.Linear",
• "transformers.models.mbart.modeling_mbart.MBartEncoderLayer",
• "torch.nn.modules.container.ModuleList",
• "Model.MLP_arg_classification",
• "torch._utils._rebuild_tensor_v2",
• "transformers.models.mbart.configuration_mbart.MBartConfig"

How to fix it?