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 /modules/sd_hijack.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 'modules/sd_hijack.py')
| -rw-r--r-- | modules/sd_hijack.py | 8 |
1 files changed, 0 insertions, 8 deletions
diff --git a/modules/sd_hijack.py b/modules/sd_hijack.py index 6b0d95af..870eba88 100644 --- a/modules/sd_hijack.py +++ b/modules/sd_hijack.py @@ -69,12 +69,6 @@ def undo_optimizations(): ldm.modules.diffusionmodules.model.AttnBlock.forward = diffusionmodules_model_AttnBlock_forward
-def fix_checkpoint():
- ldm.modules.attention.BasicTransformerBlock.forward = sd_hijack_checkpoint.BasicTransformerBlock_forward
- ldm.modules.diffusionmodules.openaimodel.ResBlock.forward = sd_hijack_checkpoint.ResBlock_forward
- ldm.modules.diffusionmodules.openaimodel.AttentionBlock.forward = sd_hijack_checkpoint.AttentionBlock_forward
-
-
class StableDiffusionModelHijack:
fixes = None
comments = []
@@ -106,8 +100,6 @@ class StableDiffusionModelHijack: self.optimization_method = apply_optimizations()
self.clip = m.cond_stage_model
-
- fix_checkpoint()
def flatten(el):
flattened = [flatten(children) for children in el.children()]
|