Mirrors
/
EveryDream2trainer
mirror of https://github.com/victorchall/EveryDream2trainer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix log leaking color

This commit is contained in:
Damian Stewart 2023-03-02 22:53:59 +01:00
parent e2fd45737d
commit 15187ae2e2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -537,7 +537,7 @@ def main(args):
logging.info(f"{Fore.CYAN} * Training Text Encoder Only *{Style.RESET_ALL}") logging.info(f"{Fore.CYAN} * Training Text Encoder Only *{Style.RESET_ALL}")
if text_encoder_lr_scale != 1: if text_encoder_lr_scale != 1:
logging.warning(f"{Fore.YELLOW} * Ignoring text_encoder_lr_scale {text_encoder_lr_scale} and using the " logging.warning(f"{Fore.YELLOW} * Ignoring text_encoder_lr_scale {text_encoder_lr_scale} and using the "
f"Unet LR {curr_lr} for the text encoder instead.") f"Unet LR {curr_lr} for the text encoder instead *{Style.RESET_ALL}")
params_to_train = itertools.chain(text_encoder.parameters()) params_to_train = itertools.chain(text_encoder.parameters())
else: else:
logging.info(f"{Fore.CYAN} * Training Text and Unet *{Style.RESET_ALL}") logging.info(f"{Fore.CYAN} * Training Text and Unet *{Style.RESET_ALL}")