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

Replaced print with logger

Browse Source
This commit is contained in:
Yuta Hayashibe
2024-02-25 00:50:36 +09:00
parent e69d34103b
commit 5d5f39b6e6
5 changed files with 138 additions and 125 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
networks/dylora.py
View File

@@ -327,10 +327,10 @@ class DyLoRANetwork(torch.nn.Module):
for i, text_encoder in enumerate(text_encoders):
if len(text_encoders) > 1:
index = i + 1
print(f"create LoRA for Text Encoder {index}")
logger.info(f"create LoRA for Text Encoder {index}")
else:
index = None
print(f"create LoRA for Text Encoder")
logger.info("create LoRA for Text Encoder")
text_encoder_loras = create_modules(False, text_encoder, DyLoRANetwork.TEXT_ENCODER_TARGET_REPLACE_MODULE)
self.text_encoder_loras.extend(text_encoder_loras)