VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

improve OFT implementation closes #944

Browse Source
This commit is contained in:
Kohya S
2024-09-13 22:37:11 +09:00
parent 43ad73860d
commit 93d9fbf607
4 changed files with 88 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
networks/check_lora_weights.py
View File

@@ -18,7 +18,7 @@ def main(file):
keys = list(sd.keys())
for key in keys:
if "lora_up" in key or "lora_down" in key:
if "lora_up" in key or "lora_down" in key or "lora_A" in key or "lora_B" in key or "oft_" in key:
values.append((key, sd[key]))
print(f"number of LoRA modules: {len(values)}")