Raise exception when failing to find diffuser model

2026-09-18 16:54:33 +02:00 · 2023-05-30 12:50:41 -07:00
parent d9f72b066f
commit 2e7aa7eb15
1 changed files with 3 additions and 2 deletions
@@ -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']