diff --git a/ui/easydiffusion/model_manager.py b/ui/easydiffusion/model_manager.py index 70bcdacf..638523db 100644 --- a/ui/easydiffusion/model_manager.py +++ b/ui/easydiffusion/model_manager.py @@ -211,6 +211,7 @@ def getModels(): listModels(model_type='stable-diffusion') listModels(model_type='vae') listModels(model_type='hypernetwork') + listModels(model_type='gfpgan') if models_scanned > 0: log.info(f'[green]Scanned {models_scanned} models. Nothing infected[/]') diff --git a/ui/index.html b/ui/index.html index 9abc0e2e..81719238 100644 --- a/ui/index.html +++ b/ui/index.html @@ -232,7 +232,7 @@