From 28f06c7200a851f6236b2d2deeaf8e2db5bc78fc Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Thu, 22 Sep 2022 00:36:53 +1000
Subject: [PATCH] Reverts divergent fix to resto module path

---
 scripts/dream.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/dream.py b/scripts/dream.py
index 9efa1829c6..036a3704c8 100755
--- a/scripts/dream.py
+++ b/scripts/dream.py
@@ -47,7 +47,7 @@ def main():
     # Loading Face Restoration and ESRGAN Modules
     try:
         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)
         if opt.restore:
             gfpgan, codeformer = restoration.load_face_restore_models()