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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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]: