theroyallab/tabbyAPI
1
0
Fork 0
mirror of https://github.com/theroyallab/tabbyAPI.git synced 2026-04-27 09:41:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Model: Fix load if model didn't load properly

Browse Source 
If the model didn't load properly, the container still exists until
unload is called. However, the name check still registered as true.

Signed-off-by: kingbri <bdashore3@proton.me>
This commit is contained in:
kingbri
2024-03-16 22:57:45 -04:00
committed by Brian Dashore
parent 2755fd1af0
commit 95e44c20d6
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
common/model.py
View File

@@ -33,7 +33,7 @@ async def load_model_gen(model_path: pathlib.Path, **kwargs):
if container and container.model: if container and container.model:
loaded_model_name = container.get_model_path().name loaded_model_name = container.get_model_path().name
if loaded_model_name == model_path.name: if loaded_model_name == model_path.name and container.model_loaded:
raise ValueError( raise ValueError(
f'Model "{loaded_model_name}" is already loaded! Aborting.' f'Model "{loaded_model_name}" is already loaded! Aborting.'
) )