We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b4aa4eb commit f7b19adCopy full SHA for f7b19ad
autointent/modules/scoring/_gcn/gcn_scorer.py
@@ -165,6 +165,6 @@ def load(
165
cross_encoder_config: CrossEncoderConfig | None = None,
166
) -> Self:
167
instance = super().load(path, embedder_config, cross_encoder_config)
168
- if hasattr(instance, "_label_embeddings"):
+ if hasattr(instance, "label_embeddings"):
169
instance.label_embeddings = torch.tensor(instance.label_embeddings).to(instance.torch_config.device)
170
return instance
0 commit comments