diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-10-13 17:12:37 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-10-13 17:12:37 +0000 |
| commit | 354ef0da3b1f0fa5c113d04b6c79e3908c848d23 (patch) | |
| tree | f3ee31caf694d83940b22d4aa1e8a5470b88079b /modules/hypernetworks | |
| parent | a10b0e11fc22cc67b6a3664f2ddd17425d8433a8 (diff) | |
| download | stable-diffusion-webui-gfx803-354ef0da3b1f0fa5c113d04b6c79e3908c848d23.tar.gz stable-diffusion-webui-gfx803-354ef0da3b1f0fa5c113d04b6c79e3908c848d23.tar.bz2 stable-diffusion-webui-gfx803-354ef0da3b1f0fa5c113d04b6c79e3908c848d23.zip | |
add hypernetwork multipliers
Diffstat (limited to 'modules/hypernetworks')
| -rw-r--r-- | modules/hypernetworks/hypernetwork.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/modules/hypernetworks/hypernetwork.py b/modules/hypernetworks/hypernetwork.py index b6c06d49..f1248bb7 100644 --- a/modules/hypernetworks/hypernetwork.py +++ b/modules/hypernetworks/hypernetwork.py @@ -18,6 +18,8 @@ from modules.textual_inversion.learn_schedule import LearnRateScheduler class HypernetworkModule(torch.nn.Module):
+ multiplier = 1.0
+
def __init__(self, dim, state_dict=None):
super().__init__()
@@ -36,7 +38,11 @@ class HypernetworkModule(torch.nn.Module): self.to(devices.device)
def forward(self, x):
- return x + (self.linear2(self.linear1(x)))
+ return x + (self.linear2(self.linear1(x))) * self.multiplier
+
+
+def apply_strength(value=None):
+ HypernetworkModule.multiplier = value if value is not None else shared.opts.sd_hypernetwork_strength
class Hypernetwork:
|