Update lora load in comfyui && Fix bug in qwen image training. (#318)

2025-09-15 13:45:54 +08:00
parent abeac38889
commit a40638f853
5 changed files with 1960 additions and 12 deletions
@@ -135,7 +135,7 @@ check_min_version("0.18.0.dev0")

 logger = get_logger(__name__, log_level="INFO")

-def log_validation(vae, text_encoder, tokenizer, transformer3d, args, config, accelerator, weight_dtype, global_step):
+def log_validation(vae, text_encoder, tokenizer, transformer3d, args, accelerator, weight_dtype, global_step):
    try:
        logger.info("Running validation... ")

@@ -1566,7 +1566,6 @@ def main():
                            tokenizer,
                            transformer3d,
                            args,
-                            config,
                            accelerator,
                            weight_dtype,
                            global_step,
@@ -1593,7 +1592,6 @@ def main():
                    tokenizer,
                    transformer3d,
                    args,
-                    config,
                    accelerator,
                    weight_dtype,
                    global_step,