add missing exception name

2024-08-30 20:32:17 +00:00 · 2023-07-16 16:14:28 -04:00 · 2023-07-16 16:14:28 -04:00 · 2bc3e36bc0
commit 2bc3e36bc0
parent dabd2bf301
1 changed files with 1 additions and 0 deletions
--- a/invokeai/backend/model_management/models/vae.py
+++ b/invokeai/backend/model_management/models/vae.py
@ -16,6 +16,7 @@ from .base import (
    calc_model_size_by_data,
    classproperty,
    InvalidModelException,
+    ModelNotFoundException,
 )
 from invokeai.app.services.config import InvokeAIAppConfig
 from diffusers.utils import is_safetensors_available