Skip to content

Commit 5566413

Browse files
authored
Merge pull request #1291 from llmware-ai/update-041426-1
update gguf mtmd free on model del
2 parents 1f14b08 + 5701af7 commit 5566413

1 file changed

Lines changed: 7 additions & 0 deletions

File tree

llmware/models.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14892,6 +14892,13 @@ def __init__(self, model_name=None, model_card=None, api_key=None, prompt_wrappe
1489214892

1489314893
self.post_init()
1489414894

14895+
def __del__(self):
14896+
14897+
logger.info(f"GGUFVisionGenerativeModel - cleaning up mtmd free on closing model instance")
14898+
14899+
if self.mtmd_ctx is not None:
14900+
self._libmtmd.mtmd_free(self.mtmd_ctx)
14901+
1489514902
def load_model_for_inference(self, model_repo_path, model_card=None, **kwargs):
1489614903

1489714904
""" Loads and instantiates model along with other required objects. """

0 commit comments

Comments
 (0)