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 6c00cff commit 61e105eCopy full SHA for 61e105e
autointent/_embedder.py
@@ -118,10 +118,11 @@ def _load_model(self) -> None:
118
119
def clear_ram(self) -> None:
120
"""Move the embedding model to CPU and delete it from memory."""
121
- self._logger.debug("Clearing embedder %s from memory", self.config.model_name)
122
- self.embedding_model.cpu()
123
- del self.embedding_model
124
- torch.cuda.empty_cache()
+ if hasattr(self, "embedding_model"):
+ self._logger.debug("Clearing embedder %s from memory", self.config.model_name)
+ self.embedding_model.cpu()
+ del self.embedding_model
125
+ torch.cuda.empty_cache()
126
127
def delete(self) -> None:
128
"""Delete the embedding model and its associated directory."""
0 commit comments