Reverts divergent fix to resto module path

This commit is contained in:
psychedelicious 2022-09-22 00:36:53 +10:00
parent 7cf7ba42fb
commit 50ac367a38

View File

@ -47,7 +47,7 @@ def main():
# Loading Face Restoration and ESRGAN Modules # Loading Face Restoration and ESRGAN Modules
try: try:
gfpgan, codeformer, esrgan = None, None, None gfpgan, codeformer, esrgan = None, None, None
from ldm.dream.restoration.base import Restoration from ldm.dream.restoration import Restoration
restoration = Restoration(opt.gfpgan_dir, opt.gfpgan_model_path, opt.esrgan_bg_tile) restoration = Restoration(opt.gfpgan_dir, opt.gfpgan_model_path, opt.esrgan_bg_tile)
if opt.restore: if opt.restore:
gfpgan, codeformer = restoration.load_face_restore_models() gfpgan, codeformer = restoration.load_face_restore_models()