aboutsummaryrefslogtreecommitdiffstats
path: root/extensions-builtin/Lora/network_ia3.py
diff options
context:
space:
mode:
authorwangshuai09 <391746016@qq.com>2024-01-30 11:15:41 +0000
committerGitHub <noreply@github.com>2024-01-30 11:15:41 +0000
commit74ff85a1a1ee4cce432b1c7d33c1eda831f68d48 (patch)
tree99b70e0fef8422c8f603bf7faa1a393091cb2a8b /extensions-builtin/Lora/network_ia3.py
parentec124607f47371a6cfd61a795f86a7f1cbd44651 (diff)
parentce168ab5dbc8b54b7245f352a2eaa55a37019b91 (diff)
downloadstable-diffusion-webui-gfx803-74ff85a1a1ee4cce432b1c7d33c1eda831f68d48.tar.gz
stable-diffusion-webui-gfx803-74ff85a1a1ee4cce432b1c7d33c1eda831f68d48.tar.bz2
stable-diffusion-webui-gfx803-74ff85a1a1ee4cce432b1c7d33c1eda831f68d48.zip
Merge branch 'dev' into npu_support
Diffstat (limited to 'extensions-builtin/Lora/network_ia3.py')
-rw-r--r--extensions-builtin/Lora/network_ia3.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/extensions-builtin/Lora/network_ia3.py b/extensions-builtin/Lora/network_ia3.py
index 7edc4249..96faeaf3 100644
--- a/extensions-builtin/Lora/network_ia3.py
+++ b/extensions-builtin/Lora/network_ia3.py
@@ -17,7 +17,7 @@ class NetworkModuleIa3(network.NetworkModule):
self.on_input = weights.w["on_input"].item()
def calc_updown(self, orig_weight):
- w = self.w.to(orig_weight.device, dtype=orig_weight.dtype)
+ w = self.w.to(orig_weight.device)
output_shape = [w.size(0), orig_weight.size(1)]
if self.on_input: