From 2e7aa7eb15d964a407fd087b09488efad599248d Mon Sep 17 00:00:00 2001 From: Alexander Brown Date: Tue, 30 May 2023 12:50:41 -0700 Subject: [PATCH] Raise exception when failing to find diffuser model --- modules/sd_models.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/modules/sd_models.py b/modules/sd_models.py index 5a5737bec..98621e90f 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -50,8 +50,9 @@ class CheckpointInfo: else: # TODO Diffusers repo = [r for r in modelloader.diffuser_repos if filename == r['filename']] if len(repo) == 0: - shared.log.error(f'Cannot find diffuser model: {filename}') - return + error_message = f'Cannot find diffuser model: {filename}' + shared.log.error(error_message) + raise ValueError(error_message) self.name = repo[0]['name'] self.hash = repo[0]['hash'][:8] self.sha256 = repo[0]['hash']