ml6team

/

mt5-small-german-query-generation

like 2

• "transformers.models.t5.modeling_t5.T5LayerFF",
• "transformers.models.t5.modeling_t5.T5DenseGatedGeluDense",
• "torch._utils._rebuild_parameter",
• "transformers.models.t5.modeling_t5.T5LayerCrossAttention",
• "torch.nn.modules.sparse.Embedding",
• "torch._utils._rebuild_tensor_v2",
• "transformers.models.mt5.modeling_mt5.MT5ForConditionalGeneration",
• "transformers.models.t5.modeling_t5.T5LayerSelfAttention",
• "torch.nn.modules.container.ModuleList",
• "transformers.models.t5.modeling_t5.T5Block",
• "collections.OrderedDict",
• "torch.FloatStorage",
• "torch.nn.modules.dropout.Dropout",
• "transformers.models.t5.modeling_t5.T5Stack",
• "transformers.activations.NewGELUActivation",
• "transformers.models.t5.modeling_t5.T5LayerNorm",
• "torch.nn.modules.linear.Linear",
• "transformers.models.mt5.configuration_mt5.MT5Config",
• "transformers.models.t5.modeling_t5.T5Attention",
• "__builtin__.set"

How to fix it?