minor fix

This commit is contained in:
mi804
2025-08-14 13:59:04 +08:00
parent 49f9a11eb3
commit 3212c83398
2 changed files with 1 additions and 2 deletions

View File

@@ -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_"):