refactor upscalers

This commit is contained in:
Vladimir Mandic
2023-09-20 13:07:29 -04:00
parent be70c0c576
commit 8ca79ad9b5
29 changed files with 328 additions and 418 deletions
+1 -1
View File
@@ -22,7 +22,6 @@ from modules.textual_inversion.preprocess import preprocess
from modules.hypernetworks.hypernetwork import create_hypernetwork, train_hypernetwork
from modules.sd_models import checkpoints_list, unload_model_weights, reload_model_weights
from modules.sd_models_config import find_checkpoint_config_near_filename
from modules.realesrgan_model import get_realesrgan_models
from modules import devices
errors.install()
@@ -478,6 +477,7 @@ class Api:
return [{"name":x.name(), "cmd_dir": getattr(x, "cmd_dir", None)} for x in shared.face_restorers]
def get_realesrgan_models(self):
from modules.postprocess.realesrgan_model import get_realesrgan_models
return [{"name":x.name,"path":x.data_path, "scale":x.scale} for x in get_realesrgan_models(None)]
def get_prompt_styles(self):