mirror of
https://github.com/gbtb/nix-stable-diffusion.git
synced 2024-11-05 04:55:20 +03:00
14 lines
409 B
Diff
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():
|