aithings/packages/realesrgan/root_dir.patch
2024-02-24 19:18:13 +01:00

14 lines
409 B
Diff

diff --git a/realesrgan/utils.py b/realesrgan/utils.py
index 67e5232..3d36793 100644
--- a/realesrgan/utils.py
+++ b/realesrgan/utils.py
@@ -8,7 +8,7 @@ import torch
from basicsr.utils.download_util import load_file_from_url
from torch.nn import functional as F
-ROOT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+ROOT_DIR = f'{os.getcwd()}/src/realesrgan'
class RealESRGANer():