diff --git a/nodes.py b/nodes.py index a38b916..9057917 100644 --- a/nodes.py +++ b/nodes.py @@ -208,7 +208,7 @@ class InitFluxTraining: "network_module": "networks.lora_flux", "dataset_config": dataset, "output_dir": output_dir, - "output_name": f"{output_name}_rank{kwargs.get["network_dim"]}_{save_dtype}", + "output_name": f"{output_name}_rank{kwargs.get("network_dim")}_{save_dtype}", "loss_type": "l2", "optimizer_type": optimizer_type, "guidance_scale": 3.5,