Download all model types.

This commit is contained in:
camenduru
2023-07-24 10:59:59 +03:00
committed by GitHub
parent 86b62cfccc
commit cbb90cbdbb

View File

@ -145,8 +145,7 @@ class ModelInstall(object):
models = set() models = set()
for key, value in self.datasets.items(): for key, value in self.datasets.items():
name,base,model_type = ModelManager.parse_key(key) name,base,model_type = ModelManager.parse_key(key)
if model_type==ModelType.Main: models.add(key)
models.add(key)
return models return models
def recommended_models(self)->Set[str]: def recommended_models(self)->Set[str]: