feat(ui): close starter models toast when a model is installed

invoke-ai · May 21, 2024 · 02887aa · 02887aa
1 parent 05659be
commit 02887aa
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/invokeai/frontend/web/src/features/modelManagerV2/hooks/useStarterModelsToast.tsx b/invokeai/frontend/web/src/features/modelManagerV2/hooks/useStarterModelsToast.tsx
@@ -17,7 +17,11 @@ export const useStarterModelsToast = () => {
 
   useEffect(() => {
     if (toast.isActive(TOAST_ID)) {
-      return;
+      if (mainModels.length === 0) {
+        return;
+      } else {
+        toast.close(TOAST_ID);
+      }
     }
     if (data && mainModels.length === 0 && !didToast && isEnabled) {
       toast({