Skip to content

Commit b062db0

Browse files
committed
Override networks.lora -> networks.lora_anima
1 parent 47fde62 commit b062db0

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

anima_train_network.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -603,4 +603,10 @@ def setup_parser() -> argparse.ArgumentParser:
603603
args = train_util.read_config_from_file(args, parser)
604604

605605
trainer = AnimaNetworkTrainer()
606+
607+
# Automatically switch to Anima-specific LoRA module if generic one is provided
608+
if args.network_module == "networks.lora":
609+
print("Override network module: networks.lora -> networks.lora_anima")
610+
args.network_module = "networks.lora_anima"
611+
606612
trainer.train(args)

0 commit comments

Comments
 (0)