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

change to use_reentrant=False

Browse Source
This commit is contained in:
ykume
2023-08-17 10:19:14 +09:00
parent 3f7235c36f
commit 306ee24c90
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
networks/lora_control_net.py
View File

@@ -5,7 +5,7 @@ from networks.lora import LoRAModule, LoRANetwork
from library import sdxl_original_unet
SKIP_OUTPUT_BLOCKS = False
SKIP_OUTPUT_BLOCKS = True
SKIP_CONV2D = False
TRANSFORMER_ONLY = True # if True, SKIP_CONV2D is ignored
ATTN1_ETC_ONLY = True
@@ -286,7 +286,7 @@ if __name__ == "__main__":
unet.to("cuda").to(torch.float16)
print("create LoRA controlnet")
control_net = LoRAControlNet(unet, 128, 32, 1)
control_net = LoRAControlNet(unet, 256, 64, 1)
control_net.apply_to()
control_net.to("cuda")

2
sdxl_train_lora_control_net.py
View File

@@ -813,7 +813,7 @@ def setup_parser() -> argparse.ArgumentParser:
if __name__ == "__main__":
# sdxl_original_unet.USE_REENTRANT = False
sdxl_original_unet.USE_REENTRANT = False
parser = setup_parser()