Skip to content

Commit ebf066b

Browse files
committed
clear ram bug fix
1 parent 714c8c2 commit ebf066b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/autointent/_wrappers/embedder.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -185,7 +185,7 @@ def train(self, utterances: list[str], labels: ListOfLabels, config: EmbedderFin
185185

186186
def clear_ram(self) -> None:
187187
"""Move the embedding model to CPU and delete it from memory."""
188-
if hasattr(self, "embedding_model"):
188+
if hasattr(self, "_model"):
189189
logger.debug("Clearing embedder %s from memory", self.config.model_name)
190190
self._model.cpu()
191191
del self._model

0 commit comments

Comments
 (0)