diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-10-07 20:22:22 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-10-07 20:22:22 +0000 |
| commit | 12c4d5c6b5bf9dd50d0601c36af4f99b65316d58 (patch) | |
| tree | 30bbd10c806afd8168cbc0cd0f33529bde8d1f47 /modules/sd_hijack.py | |
| parent | f7c787eb7c295c27439f4fbdf78c26b8389560be (diff) | |
| download | stable-diffusion-webui-gfx803-12c4d5c6b5bf9dd50d0601c36af4f99b65316d58.tar.gz stable-diffusion-webui-gfx803-12c4d5c6b5bf9dd50d0601c36af4f99b65316d58.tar.bz2 stable-diffusion-webui-gfx803-12c4d5c6b5bf9dd50d0601c36af4f99b65316d58.zip | |
hypernetwork training mk1
Diffstat (limited to 'modules/sd_hijack.py')
| -rw-r--r-- | modules/sd_hijack.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/modules/sd_hijack.py b/modules/sd_hijack.py index d68f89cc..ec8c9d4b 100644 --- a/modules/sd_hijack.py +++ b/modules/sd_hijack.py @@ -8,7 +8,7 @@ from torch import einsum from torch.nn.functional import silu
import modules.textual_inversion.textual_inversion
-from modules import prompt_parser, devices, sd_hijack_optimizations, shared, hypernetwork
+from modules import prompt_parser, devices, sd_hijack_optimizations, shared
from modules.shared import opts, device, cmd_opts
import ldm.modules.attention
@@ -32,6 +32,8 @@ def apply_optimizations(): def undo_optimizations():
+ from modules.hypernetwork import hypernetwork
+
ldm.modules.attention.CrossAttention.forward = hypernetwork.attention_CrossAttention_forward
ldm.modules.diffusionmodules.model.nonlinearity = diffusionmodules_model_nonlinearity
ldm.modules.diffusionmodules.model.AttnBlock.forward = diffusionmodules_model_AttnBlock_forward
|