ISRI/ai-server
10
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

mark the model as loaded once loaded

Browse source
This commit is contained in:
faraphel 2025-01-08 20:35:14 +01:00
parent 66ae63ac19
commit 71eff944dd
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
source/model/base/BaseModel.py
View file

@ -64,6 +64,9 @@ class BaseModel(abc.ABC):
# declare ourselves as the currently loaded model # declare ourselves as the currently loaded model
self.manager.current_loaded_model = self self.manager.current_loaded_model = self
# mark the model as loaded
self._loaded = True
@abc.abstractmethod @abc.abstractmethod
def _load(self): def _load(self):
""" """