search

ostris / ai-toolkit

Various AI scripts. Mostly Stable Diffusion stuff.
MIT License
220 stars 25 forks source link

[Development] Mismatch of tensor dimension offset multiplier #18

Open KappaMund opened 9 months ago

KappaMund commented 9 months ago

======================================== Traceback (most recent call last): File "C:!NeuralNetwork\ai-toolkit\run.py", line 86, in main() File "C:!NeuralNetwork\ai-toolkit\run.py", line 82, in main raise e File "C:!NeuralNetwork\ai-toolkit\run.py", line 74, in main job.run() File "C:!NeuralNetwork\ai-toolkit\jobs\TrainJob.py", line 49, in run process.run() File "C:!NeuralNetwork\ai-toolkit\jobs\process\BaseSDTrainProcess.py", line 950, in run loss_dict = self.hook_train_loop(batch) ^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "C:!NeuralNetwork\ai-toolkit\jobs\process\TrainSliderProcess.py", line 553, in hook_train_loop offset_multiplier = offset_multiplier.view(offset.shape[0], 1, 1, 1) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ RuntimeError: shape '[6, 1, 1, 1]' is invalid for input of size 1

You can fix the error by changing offset_multiplier = offset_multiplier.view(offset.shape[0], 1, 1, 1) on offset_multiplier = offset_multiplier.expand(offset.shape[0], 1, 1, 1)