better handle partial models

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2024-10-24 11:36:21 -04:00
parent f4ed53c159
commit 321bfe8bc7
5 changed files with 10 additions and 8 deletions
@@ -262,7 +262,6 @@ class EmbeddingDatabase:
if shared.native:
return 0
if not shared.sd_loaded:
shared.log.error('Model not loaded')
return 0
vec = shared.sd_model.cond_stage_model.encode_embedding_init_text(",", 1)
return vec.shape[1]