diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-01-28 09:24:40 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-01-28 09:24:40 +0300 |
| commit | ce72af87d3b05c946bc82033786fc340f1c20512 (patch) | |
| tree | e02a69bc9b18f36db76590bd6312320685901b5e /modules/upscaler.py | |
| parent | 0834d4ce374225131e025540220c727e352a3e43 (diff) | |
| parent | 23a9d5e27390846dea0895a02c04aec9583a4d38 (diff) | |
Merge pull request #7199 from maxaudron/feature/configurable-data-dir
Add flag to store user data sepperate from source code
Diffstat (limited to 'modules/upscaler.py')
| -rw-r--r-- | modules/upscaler.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/modules/upscaler.py b/modules/upscaler.py index a5bf5acb..e2eaa730 100644 --- a/modules/upscaler.py +++ b/modules/upscaler.py @@ -11,7 +11,6 @@ from modules import modelloader, shared LANCZOS = (Image.Resampling.LANCZOS if hasattr(Image, 'Resampling') else Image.LANCZOS) NEAREST = (Image.Resampling.NEAREST if hasattr(Image, 'Resampling') else Image.NEAREST) -from modules.paths import models_path class Upscaler: @@ -39,7 +38,7 @@ class Upscaler: self.mod_scale = None if self.model_path is None and self.name: - self.model_path = os.path.join(models_path, self.name) + self.model_path = os.path.join(shared.models_path, self.name) if self.model_path and create_dirs: os.makedirs(self.model_path, exist_ok=True) @@ -143,4 +142,4 @@ class UpscalerNearest(Upscaler): def __init__(self, dirname=None): super().__init__(False) self.name = "Nearest" - self.scalers = [UpscalerData("Nearest", None, self)]
\ No newline at end of file + self.scalers = [UpscalerData("Nearest", None, self)] |