diff options
| author | Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com> | 2023-07-18 02:45:42 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-07-18 02:45:42 +0000 |
| commit | 3d31caf4a53c4bb4469b72790b459eba7b251da9 (patch) | |
| tree | 3c54c366b3641ea426764926df649ab0f326ebdf | |
| parent | 17e14ed2d9451859325d275ccc6cdf51fc85a56d (diff) | |
| download | stable-diffusion-webui-gfx803-3d31caf4a53c4bb4469b72790b459eba7b251da9.tar.gz stable-diffusion-webui-gfx803-3d31caf4a53c4bb4469b72790b459eba7b251da9.tar.bz2 stable-diffusion-webui-gfx803-3d31caf4a53c4bb4469b72790b459eba7b251da9.zip | |
use "is not None" for Tensor
| -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 3a94f3e9..340acdab 100644 --- a/extensions-builtin/Lora/network_lokr.py +++ b/extensions-builtin/Lora/network_lokr.py @@ -28,11 +28,11 @@ class NetworkModuleLokr(network.NetworkModule): self.w1 = weights.w.get("lokr_w1")
self.w1a = weights.w.get("lokr_w1_a")
self.w1b = weights.w.get("lokr_w1_b")
- self.dim = self.w1b.shape[0] if self.w1b else self.dim
+ self.dim = self.w1b.shape[0] if self.w1b is not None else self.dim
self.w2 = weights.w.get("lokr_w2")
self.w2a = weights.w.get("lokr_w2_a")
self.w2b = weights.w.get("lokr_w2_b")
- self.dim = self.w2b.shape[0] if self.w2b else self.dim
+ self.dim = self.w2b.shape[0] if self.w2b is not None else self.dim
self.t2 = weights.w.get("lokr_t2")
def calc_updown(self, orig_weight):
|