hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-08 13:19:54 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

add option to show/hide warnings

Browse Source 
removed hiding warnings from LDSR
fixed/reworked few places that produced warnings
This commit is contained in:
AUTOMATIC
2023-01-18 23:04:24 +03:00
parent 889b851a52
commit 924e222004
10 changed files with 71 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
extensions-builtin/LDSR/ldsr_model_arch.py
View File

@@ -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