operated in the same way as the previous version of AI toolkit, but Control image is required for Flux Kontext model generation.

[song-jz]

##Pulling the latest version of AI toolkit to train kontext always reports missing control images, provided that the control image folder has been correctly imported and operated in the same way as the previous version of AI toolkit

Error running job: Control image is required for Flux Kontext model generation.

Result:

• 0 completed jobs
• 1 failure
  ========================================
  Traceback (most recent call last):
  Traceback (most recent call last):
  File "/root/ai-toolkit/run.py", line 120, in
  File "/root/ai-toolkit/run.py", line 120, in
  main()main()
  File "/root/ai-toolkit/run.py", line 108, in main
  File "/root/ai-toolkit/run.py", line 108, in main
  raise eraise e
  File "/root/ai-toolkit/run.py", line 96, in main
  File "/root/ai-toolkit/run.py", line 96, in main
  job.run()job.run()
  File "/root/ai-toolkit/jobs/ExtensionJob.py", line 22, in run
  File "/root/ai-toolkit/jobs/ExtensionJob.py", line 22, in run
  process.run()process.run()
  File "/root/ai-toolkit/jobs/process/BaseSDTrainProcess.py", line 1932, in run
  File "/root/ai-toolkit/jobs/process/BaseSDTrainProcess.py", line 1932, in run
  self.sample(self.step_num)self.sample(self.step_num)
  File "/root/ai-toolkit/extensions_built_in/sd_trainer/UITrainer.py", line 225, in sample
  File "/root/ai-toolkit/extensions_built_in/sd_trainer/UITrainer.py", line 225, in sample
  super().sample(step, is_first)super().sample(step, is_first)
  File "/root/ai-toolkit/jobs/process/BaseSDTrainProcess.py", line 344, in sample
  File "/root/ai-toolkit/jobs/process/BaseSDTrainProcess.py", line 344, in sample
  self.sd.generate_images(gen_img_config_list, sampler=sample_config.sampler)self.sd.generate_images(gen_img_config_list, sampler=sample_config.sampler)
  File "/usr/local/lib/python3.12/dist-packages/torch/utils/_contextlib.py", line 116, in decorate_context
  File "/usr/local/lib/python3.12/dist-packages/torch/utils/_contextlib.py", line 116, in decorate_context
  return func(*args, **kwargs)return func(*args, **kwargs)
  ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/root/ai-toolkit/toolkit/models/base_model.py", line 567, in generate_images
  File "/root/ai-toolkit/toolkit/models/base_model.py", line 567, in generate_images
  img = self.generate_single_image(img = self.generate_single_image(
  ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/root/ai-toolkit/extensions_built_in/diffusion_models/flux_kontext/flux_kontext.py", line 214, in generate_single_image
  File "/root/ai-toolkit/extensions_built_in/diffusion_models/flux_kontext/flux_kontext.py", line 214, in generate_single_image
  raise ValueError(raise ValueError(
  ValueErrorValueError: : Control image is required for Flux Kontext model generation.Control image is required for Flux Kontext model generation.