diff options
| author | 不会画画的中医不是好程序员 <yfszzx@gmail.com> | 2022-10-13 23:35:07 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-10-13 23:35:07 +0000 |
| commit | 7c8903367c350e7e3ba8f90679890566446f36b2 (patch) | |
| tree | a949c12c2a93c4e20aff30adf4412a450f6b2b8e /modules/hypernetworks/hypernetwork.py | |
| parent | a1489f94283c07824a7a58353c03dc89541bbe49 (diff) | |
| parent | 08b3f7aef15f74f4d2254b1274dd66fcc7940348 (diff) | |
| download | stable-diffusion-webui-gfx803-7c8903367c350e7e3ba8f90679890566446f36b2.tar.gz stable-diffusion-webui-gfx803-7c8903367c350e7e3ba8f90679890566446f36b2.tar.bz2 stable-diffusion-webui-gfx803-7c8903367c350e7e3ba8f90679890566446f36b2.zip | |
Merge branch 'AUTOMATIC1111:master' into master
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:
|