diff options
| author | DepFA <35278260+dfaker@users.noreply.github.com> | 2022-10-10 00:38:54 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-10-10 00:38:54 +0100 |
| commit | 4117afff11c7b0a2162c73ea02be8cfa30d02640 (patch) | |
| tree | af26f1b0c9eac8c024d2e51ec8fb5ca4a4d45ed3 /modules/esrgan_model.py | |
| parent | e2c2925eb4d634b186de2c76798162ec56e2f869 (diff) | |
| parent | 45fbd1c5fec887988ab555aac75a999d4f3aff40 (diff) | |
Merge branch 'master' into embed-embeddings-in-images
Diffstat (limited to 'modules/esrgan_model.py')
| -rw-r--r-- | modules/esrgan_model.py | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/modules/esrgan_model.py b/modules/esrgan_model.py index 28548124..46ad0da3 100644 --- a/modules/esrgan_model.py +++ b/modules/esrgan_model.py @@ -5,9 +5,8 @@ import torch from PIL import Image
from basicsr.utils.download_util import load_file_from_url
-import modules.esrgam_model_arch as arch
+import modules.esrgan_model_arch as arch
from modules import shared, modelloader, images, devices
-from modules.paths import models_path
from modules.upscaler import Upscaler, UpscalerData
from modules.shared import opts
@@ -76,7 +75,6 @@ class UpscalerESRGAN(Upscaler): self.model_name = "ESRGAN_4x"
self.scalers = []
self.user_path = dirname
- self.model_path = os.path.join(models_path, self.name)
super().__init__()
model_paths = self.find_models(ext_filter=[".pt", ".pth"])
scalers = []
|