From cbb90cbdbbde52f0d932e221ea30357989c871ed Mon Sep 17 00:00:00 2001
From: camenduru <54370274+camenduru@users.noreply.github.com>
Date: Mon, 24 Jul 2023 10:59:59 +0300
Subject: [PATCH] Download all model types.

---
 invokeai/backend/install/model_install_backend.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/invokeai/backend/install/model_install_backend.py b/invokeai/backend/install/model_install_backend.py
index 0537d6069a..e4a24c76b4 100644
--- a/invokeai/backend/install/model_install_backend.py
+++ b/invokeai/backend/install/model_install_backend.py
@@ -145,8 +145,7 @@ class ModelInstall(object):
         models = set()
         for key, value in self.datasets.items():
             name,base,model_type = ModelManager.parse_key(key)
-            if model_type==ModelType.Main:
-                models.add(key)
+            models.add(key)
         return models
 
     def recommended_models(self)->Set[str]: