diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-02-19 12:16:58 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-02-19 12:16:58 +0300 |
| commit | b63a13c5ed99f7689031baff5cbd0e24a3939318 (patch) | |
| tree | beb120aec08e15d28fc39827aa9e655795ed4c9f /modules | |
| parent | e287d9b2940b13c2d19dd59750e8cde7e53cbbd8 (diff) | |
| parent | 3ca41dbded119509bb66d7952d403091a5543257 (diff) | |
Merge pull request #7651 from vladmandic/missing-imports
Add missing imports
Diffstat (limited to 'modules')
| -rw-r--r-- | modules/esrgan_model_arch.py | 1 | ||||
| -rw-r--r-- | modules/sd_hijack_inpainting.py | 1 |
2 files changed, 2 insertions, 0 deletions
diff --git a/modules/esrgan_model_arch.py b/modules/esrgan_model_arch.py index bc9ceb2a..1b52b0f5 100644 --- a/modules/esrgan_model_arch.py +++ b/modules/esrgan_model_arch.py @@ -1,5 +1,6 @@ # this file is adapted from https://github.com/victorca25/iNNfer
+from collections import OrderedDict
import math
import functools
import torch
diff --git a/modules/sd_hijack_inpainting.py b/modules/sd_hijack_inpainting.py index 478cd499..55a2ce4d 100644 --- a/modules/sd_hijack_inpainting.py +++ b/modules/sd_hijack_inpainting.py @@ -11,6 +11,7 @@ import ldm.models.diffusion.plms from ldm.models.diffusion.ddpm import LatentDiffusion from ldm.models.diffusion.plms import PLMSSampler from ldm.models.diffusion.ddim import DDIMSampler, noise_like +from ldm.models.diffusion.sampling_util import norm_thresholding @torch.no_grad() |