Traceback (most recent call last):
File "/tools/OneTrainer/scripts/train.py", line 33, in
main()
File "/tools/OneTrainer/scripts/train.py", line 24, in main
trainer.train()
File "/tools/OneTrainer/modules/trainer/GenericTrainer.py", line 505, in train
loss = self.model_setup.calculate_loss(self.model, batch, model_output_data, self.args)
File "/tools/OneTrainer/modules/modelSetup/BaseStableDiffusionXLSetup.py", line 379, in calculate_loss
return self._diffusion_loss(
File "/tools/OneTrainer/modules/modelSetup/mixin/ModelSetupDiffusionLossMixin.py", line 34, in _diffusion_loss
if data['loss_type'] == 'align_prop':
KeyError: 'loss_type'
train.txt