rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Lazyload: Safetensors

Browse Source
This commit is contained in:
somebody
2023-04-02 15:40:34 -05:00
parent 91bb433b5f
commit 9d70646e4d
6 changed files with 419 additions and 105 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modeling/inference_model.py
View File

@@ -172,9 +172,13 @@ class InferenceModel:
def load(self, save_model: bool = False, initial_load: bool = False) -> None:
"""User-facing load function. Do not override this; try `_load()` instead."""
self._pre_load()
self._load(save_model=save_model, initial_load=initial_load)
self._post_load()
def _pre_load(self) -> None:
"""Pre load hook. Called before `_load()`."""
def _post_load(self) -> None:
"""Post load hook. Called after `_load()`."""