Error running job: The size of tensor a (36) must match the size of tensor b (16) at non-singleton dimension 1

[mastermk]

Here it is again, always the same mistake: Error running job: The size of tensor a (36) must match the size of tensor b (16) at non-singleton dimension 1

======================================== Result:

• 0 completed jobs
• 1 failure ======================================== Traceback (most recent call last): File "S:\ai_toolkit\ai-toolkit\run.py", line 120, in main() File "S:\ai_toolkit\ai-toolkit\run.py", line 108, in main raise e File "S:\ai_toolkit\ai-toolkit\run.py", line 96, in main job.run() File "S:\ai_toolkit\ai-toolkit\jobs\ExtensionJob.py", line 22, in run process.run() File "S:\ai_toolkit\ai-toolkit\jobs\process\BaseSDTrainProcess.py", line 2010, in run self.sample(self.step_num) File "S:\ai_toolkit\ai-toolkit\extensions_built_in\sd_trainer\DiffusionTrainer.py", line 288, in sample super().sample(step, is_first) File "S:\ai_toolkit\ai-toolkit\jobs\process\BaseSDTrainProcess.py", line 370, in sample self.sd.generate_images(gen_img_config_list, sampler=sample_config.sampler) File "S:\ai_toolkit\ai-toolkit\venv\lib\site-packages\torch\utils_contextlib.py", line 120, in decorate_context return func(*args, **kwargs) File "S:\ai_toolkit\ai-toolkit\toolkit\models\base_model.py", line 605, in generate_images img = self.generate_single_image( File "S:\ai_toolkit\ai-toolkit\extensions_built_in\diffusion_models\wan22\wan22_14b_i2v_model.py", line 83, in generate_single_image output = pipeline( File "S:\ai_toolkit\ai-toolkit\extensions_built_in\diffusion_models\wan22\wan22_pipeline.py", line 262, in call latents = self.scheduler.step( File "S:\ai_toolkit\ai-toolkit\venv\lib\site-packages\diffusers\schedulers\scheduling_unipc_multistep.py", line 992, in step model_output_convert = self.convert_model_output(model_output, sample=sample) File "S:\ai_toolkit\ai-toolkit\venv\lib\site-packages\diffusers\schedulers\scheduling_unipc_multistep.py", line 628, in convert_model_output x0_pred = sample - sigma_t * model_output RuntimeError: The size of tensor a (36) must match the size of tensor b (16) at non-singleton dimension 1

[FreelancerDDD]

I'll be having the exact same issue until the latest pull, they just created a whole new set of problems even before this error comes up!!

[DavidJBarnes]

Is there a fix for this?

Unloading text encoder 0%| | 0/25 [00:01<?, ?it/s] Error running job: The size of tensor a (36) must match the size of tensor b (16) at non-singleton dimension 1

Result:

• 0 completed jobs
• 1 failure ======================================== Traceback (most recent call last): Traceback (most recent call last): File "/app/ai-toolkit/run.py", line 120, in File "/app/ai-toolkit/run.py", line 120, in main()main() File "/app/ai-toolkit/run.py", line 108, in main File "/app/ai-toolkit/run.py", line 108, in main raise eraise e File "/app/ai-toolkit/run.py", line 96, in main File "/app/ai-toolkit/run.py", line 96, in main job.run()job.run() File "/app/ai-toolkit/jobs/ExtensionJob.py", line 22, in run File "/app/ai-toolkit/jobs/ExtensionJob.py", line 22, in run process.run()process.run() File "/app/ai-toolkit/jobs/process/BaseSDTrainProcess.py", line 2014, in run File "/app/ai-toolkit/jobs/process/BaseSDTrainProcess.py", line 2014, in run self.sample(self.step_num)self.sample(self.step_num) File "/app/ai-toolkit/extensions_built_in/sd_trainer/DiffusionTrainer.py", line 306, in sample File "/app/ai-toolkit/extensions_built_in/sd_trainer/DiffusionTrainer.py", line 306, in sample super().sample(step, is_first)super().sample(step, is_first) File "/app/ai-toolkit/jobs/process/BaseSDTrainProcess.py", line 368, in sample File "/app/ai-toolkit/jobs/process/BaseSDTrainProcess.py", line 368, 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.10/dist-packages/torch/utils/_contextlib.py", line 124, in decorate_context File "/usr/local/lib/python3.10/dist-packages/torch/utils/_contextlib.py", line 124, in decorate_context return func(*args, **kwargs)return func(*args, **kwargs) File "/app/ai-toolkit/toolkit/models/base_model.py", line 645, in generate_images File "/app/ai-toolkit/toolkit/models/base_model.py", line 645, in generate_images img = self.generate_single_image(img = self.generate_single_image( File "/app/ai-toolkit/extensions_built_in/diffusion_models/wan22/wan22_14b_i2v_model.py", line 83, in generate_single_image File "/app/ai-toolkit/extensions_built_in/diffusion_models/wan22/wan22_14b_i2v_model.py", line 83, in generate_single_image output = pipeline(output = pipeline( File "/app/ai-toolkit/extensions_built_in/diffusion_models/wan22/wan22_pipeline.py", line 266, in call File "/app/ai-toolkit/extensions_built_in/diffusion_models/wan22/wan22_pipeline.py", line 266, in call latents = self.scheduler.step(latents = self.scheduler.step( File "/usr/local/lib/python3.10/dist-packages/diffusers/schedulers/scheduling_unipc_multistep.py", line 1088, in step File "/usr/local/lib/python3.10/dist-packages/diffusers/schedulers/scheduling_unipc_multistep.py", line 1088, in step model_output_convert = self.convert_model_output(model_output, sample=sample)model_output_convert = self.convert_model_output(model_output, sample=sample) File "/usr/local/lib/python3.10/dist-packages/diffusers/schedulers/scheduling_unipc_multistep.py", line 705, in convert_model_output File "/usr/local/lib/python3.10/dist-packages/diffusers/schedulers/scheduling_unipc_multistep.py", line 705, in convert_model_output x0_pred = sample - sigma_t * model_outputx0_pred = sample - sigma_t * model_output RuntimeErrorRuntimeError: : The size of tensor a (36) must match the size of tensor b (16) at non-singleton dimension 1The size of tensor a (36) must match the size of tensor b (16) at non-singleton dimension 1