diff options
| author | yfszzx <yfszzx@gmail.com> | 2022-10-14 03:51:26 +0000 |
|---|---|---|
| committer | yfszzx <yfszzx@gmail.com> | 2022-10-14 03:51:26 +0000 |
| commit | d48f3470c8a86f22dce106e7d3c1803e5ba92085 (patch) | |
| tree | 2526ceaffa0177b40cecc30854f66a524cf6ac71 /modules/hypernetworks/hypernetwork.py | |
| parent | 4a37c7eedeab579efec03e8dae3f3f9fd4a37b02 (diff) | |
| parent | 7c8903367c350e7e3ba8f90679890566446f36b2 (diff) | |
| download | stable-diffusion-webui-gfx803-d48f3470c8a86f22dce106e7d3c1803e5ba92085.tar.gz stable-diffusion-webui-gfx803-d48f3470c8a86f22dce106e7d3c1803e5ba92085.tar.bz2 stable-diffusion-webui-gfx803-d48f3470c8a86f22dce106e7d3c1803e5ba92085.zip | |
Merge branch 'master' of https://github.com/yfszzx/stable-diffusion-webui-plus
Diffstat (limited to 'modules/hypernetworks/hypernetwork.py')
| -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:
|