PLMS runs smoothly, DDIM fails

[capesidereal]

trafficstars

when using diffusion_sampling_mode = 'plms' everything works when using diffusion_sampling_mode = 'ddim' i get error below :

TypeError Traceback (most recent call last) Input In [27], in <cell line: 217>() 216 torch.cuda.empty_cache() 217 try: --> 218 do_run() 219 except KeyboardInterrupt: 220 pass

Input In [5], in do_run() 648 init = regen_perlin() 650 if args.diffusion_sampling_mode == 'ddim': --> 651 samples = sample_fn( 652 model, 653 (batch_size, 3, args.side_y, args.side_x), 654 clip_denoised=clip_denoised, 655 model_kwargs={}, 656 cond_fn=cond_fn, 657 progress=True, 658 skip_timesteps=skip_steps, 659 init_image=init, 660 randomize_class=randomize_class, 661 eta=eta, 662 transformation_fn=symmetry_transformation_fn, 663 transformation_percent=args.transformation_percent 664 ) 665 else: 666 samples = sample_fn( 667 model, 668 (batch_size, 3, args.side_y, args.side_x), (...) 676 order=2, 677 )

TypeError: ddim_sample_loop_progressive() got an unexpected keyword argument 'transformation_fn'