From 3212c833980ba6660c0caa6ac077a10c619a55a2 Mon Sep 17 00:00:00 2001 From: mi804 <1576993271@qq.com> Date: Thu, 14 Aug 2025 13:59:04 +0800 Subject: [PATCH] minor fix --- examples/qwen_image/model_training/train.py | 1 - examples/wanvideo/README_zh.md | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/examples/qwen_image/model_training/train.py b/examples/qwen_image/model_training/train.py index 8fb3bf8..56db9e2 100644 --- a/examples/qwen_image/model_training/train.py +++ b/examples/qwen_image/model_training/train.py @@ -82,7 +82,6 @@ class QwenImageTrainingModule(DiffusionTrainingModule): # Extra inputs controlnet_input = {} for extra_input in self.extra_inputs: - inputs_shared[extra_input] = data[extra_input] if extra_input.startswith("blockwise_controlnet_"): controlnet_input[extra_input.replace("blockwise_controlnet_", "")] = data[extra_input] elif extra_input.startswith("controlnet_"): diff --git a/examples/wanvideo/README_zh.md b/examples/wanvideo/README_zh.md index 89c539e..bcc076f 100644 --- a/examples/wanvideo/README_zh.md +++ b/examples/wanvideo/README_zh.md @@ -296,12 +296,12 @@ Wan 系列模型训练通过统一的 [`./model_training/train.py`](./model_trai * `--remove_prefix_in_ckpt`: 在 ckpt 中移除前缀。 * `--save_steps`: 保存模型的间隔 step 数量，如果设置为 None ，则每个 epoch 保存一次 * `--find_unused_parameters`: DDP 训练中是否存在未使用的参数 - * `--lora_checkpoint`: LoRA 检查点的路径。如果提供此路径，LoRA 将从此检查点加载。 * 可训练模块 * `--trainable_models`: 可训练的模型，例如 dit、vae、text_encoder。 * `--lora_base_model`: LoRA 添加到哪个模型上。 * `--lora_target_modules`: LoRA 添加到哪一层上。 * `--lora_rank`: LoRA 的秩（Rank）。 + * `--lora_checkpoint`: LoRA 检查点的路径。如果提供此路径，LoRA 将从此检查点加载。 * 额外模型输入 * `--extra_inputs`: 额外的模型输入，以逗号分隔。 * 显存管理