diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-01-18 20:04:24 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-01-18 20:04:24 +0000 |
| commit | 924e222004ab54273806c5f2ca7a0e7cfa76ad83 (patch) | |
| tree | 153a08105ee2bc87df43a8a1423df96d25a8e19b /extensions-builtin/LDSR/ldsr_model_arch.py | |
| parent | 889b851a5260ce869a3286ad15d17d1bbb1da0a7 (diff) | |
| download | stable-diffusion-webui-gfx803-924e222004ab54273806c5f2ca7a0e7cfa76ad83.tar.gz stable-diffusion-webui-gfx803-924e222004ab54273806c5f2ca7a0e7cfa76ad83.tar.bz2 stable-diffusion-webui-gfx803-924e222004ab54273806c5f2ca7a0e7cfa76ad83.zip | |
add option to show/hide warnings
removed hiding warnings from LDSR
fixed/reworked few places that produced warnings
Diffstat (limited to 'extensions-builtin/LDSR/ldsr_model_arch.py')
| -rw-r--r-- | extensions-builtin/LDSR/ldsr_model_arch.py | 3 |
1 files changed, 0 insertions, 3 deletions
diff --git a/extensions-builtin/LDSR/ldsr_model_arch.py b/extensions-builtin/LDSR/ldsr_model_arch.py index 0ad49f4e..bc11cc6e 100644 --- a/extensions-builtin/LDSR/ldsr_model_arch.py +++ b/extensions-builtin/LDSR/ldsr_model_arch.py @@ -1,7 +1,6 @@ import os import gc import time -import warnings import numpy as np import torch @@ -15,8 +14,6 @@ from ldm.models.diffusion.ddim import DDIMSampler from ldm.util import instantiate_from_config, ismap from modules import shared, sd_hijack -warnings.filterwarnings("ignore", category=UserWarning) - cached_ldsr_model: torch.nn.Module = None |