| 348 | self.guidance_optimization_params = guidance_optimization_params |
| 349 | |
| 350 | def set_diffusion_specific_params(self, diffusion_specific_params): |
| 351 | self.apply_guidance_intermediate = diffusion_specific_params['apply_guidance_intermediate'] |
| 352 | self.apply_guidance_output = diffusion_specific_params['apply_guidance_output'] |
| 353 | self.final_step_opt_params = diffusion_specific_params['final_step_opt_params'] |
| 354 | self.stride = diffusion_specific_params['stride'] |
| 355 | |
| 356 | #------------------------------------------ utility ------------------------------------------# |
| 357 | def _create_dynamics(self): |