weizhou03

/

ModernBERT-base

like 0

• "torch.device",
• "collections.OrderedDict",
• "torch.FloatStorage",
• "torch._utils._rebuild_tensor_v2"

How to fix it?

• "transformers.activations.GELUActivation",
• "torch._utils._rebuild_tensor_v2",
• "collections.OrderedDict",
• "transformers.models.modernbert.modeling_modernbert.ModernBertEmbeddings",
• "torch.nn.modules.container.ModuleList",
• "transformers.models.modernbert.modeling_modernbert.ModernBertMLP",
• "transformers.models.modernbert.modeling_modernbert.ModernBertEncoderLayer",
• "transformers.modeling_rope_utils._compute_default_rope_parameters",
• "torch.float32",
• "__builtin__.set",
• "assistant_model.ModernBERTMultiLabelClassifier",
• "torch.nn.modules.sparse.Embedding",
• "transformers.models.modernbert.configuration_modernbert.ModernBertConfig",
• "torch._C._nn.gelu",
• "torch._utils._rebuild_parameter",
• "torch.nn.modules.linear.Identity",
• "torch.nn.modules.dropout.Dropout",
• "transformers.models.modernbert.modeling_modernbert.ModernBertModel",
• "torch.FloatStorage",
• "transformers.models.modernbert.modeling_modernbert.ModernBertAttention",
• "transformers.models.modernbert.modeling_modernbert.ModernBertRotaryEmbedding",
• "torch.nn.modules.normalization.LayerNorm",
• "torch.nn.modules.linear.Linear"

How to fix it?