diff options
| author | Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com> | 2023-07-18 10:23:41 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-07-18 10:23:41 +0800 |
| commit | 17e14ed2d9451859325d275ccc6cdf51fc85a56d (patch) | |
| tree | 0553796fbaab0b3b201598c36b1a1bebe822e70b /extensions-builtin/Lora/network_lokr.py | |
| parent | a99d5708e6d603e8f7cfd1b8c6595f8026219ba0 (diff) | |
Fix wrong key name in lokr module
Diffstat (limited to 'extensions-builtin/Lora/network_lokr.py')
| -rw-r--r-- | extensions-builtin/Lora/network_lokr.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/extensions-builtin/Lora/network_lokr.py b/extensions-builtin/Lora/network_lokr.py index 920062e2..3a94f3e9 100644 --- a/extensions-builtin/Lora/network_lokr.py +++ b/extensions-builtin/Lora/network_lokr.py @@ -6,8 +6,8 @@ import network class ModuleTypeLokr(network.ModuleType):
def create_module(self, net: network.Network, weights: network.NetworkWeights):
- has_1 = "lokr_w1" in weights.w or ("lokr_w1a" in weights.w and "lokr_w1b" in weights.w)
- has_2 = "lokr_w2" in weights.w or ("lokr_w2a" in weights.w and "lokr_w2b" in weights.w)
+ has_1 = "lokr_w1" in weights.w or ("lokr_w1_a" in weights.w and "lokr_w1_b" in weights.w)
+ has_2 = "lokr_w2" in weights.w or ("lokr_w2_a" in weights.w and "lokr_w2_b" in weights.w)
if has_1 and has_2:
return NetworkModuleLokr(net, weights)
|