diff options
| author | flamelaw <flamelaw.com3d2@gmail.com> | 2022-11-23 11:21:52 +0000 |
|---|---|---|
| committer | flamelaw <flamelaw.com3d2@gmail.com> | 2022-11-23 11:21:52 +0000 |
| commit | 1bd57cc9791e2e742f72a3d74d589f2c289e8e92 (patch) | |
| tree | d76b8c6087aac5120ed7760905ca7c0913f25431 | |
| parent | d2c97fc3fe5857d6fba9ad1695ed3ac6ec455ca9 (diff) | |
| download | stable-diffusion-webui-gfx803-1bd57cc9791e2e742f72a3d74d589f2c289e8e92.tar.gz stable-diffusion-webui-gfx803-1bd57cc9791e2e742f72a3d74d589f2c289e8e92.tar.bz2 stable-diffusion-webui-gfx803-1bd57cc9791e2e742f72a3d74d589f2c289e8e92.zip | |
last_layer_dropout default to False
| -rw-r--r-- | modules/hypernetworks/hypernetwork.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/hypernetworks/hypernetwork.py b/modules/hypernetworks/hypernetwork.py index 9388959f..8466887f 100644 --- a/modules/hypernetworks/hypernetwork.py +++ b/modules/hypernetworks/hypernetwork.py @@ -38,7 +38,7 @@ class HypernetworkModule(torch.nn.Module): activation_dict.update({cls_name.lower(): cls_obj for cls_name, cls_obj in inspect.getmembers(torch.nn.modules.activation) if inspect.isclass(cls_obj) and cls_obj.__module__ == 'torch.nn.modules.activation'})
def __init__(self, dim, state_dict=None, layer_structure=None, activation_func=None, weight_init='Normal',
- add_layer_norm=False, use_dropout=False, activate_output=False, last_layer_dropout=True):
+ add_layer_norm=False, use_dropout=False, activate_output=False, last_layer_dropout=False):
super().__init__()
assert layer_structure is not None, "layer_structure must not be None"
|