search

ermongroup / SDEdit

PyTorch implementation for SDEdit: Image Synthesis and Editing with Stochastic Differential Equations
https://sde-image-editing.github.io/
MIT License
966 stars 90 forks source link

How to change the step length or the total denoising steps (N), so that the image generation process could be faster? #12

Open yutxie opened 2 years ago

yutxie commented 2 years ago

Hi there, thank you for releasing the code!

I tried several ways as below to change the step length (delta t) and the total denosing steps but none of them works:

  1. Changing the num_timesteps hyperparameter in the config file and changing thetotal_noise_levels accordingly.
  2. Changing the step length of the enumeration of i in the SDEditing demonstration function

It seems the synthesized images remain noised after the hyperparameters are tuned. Is there a way to modify N safely? Or is it possible to accelerate the generation process by other means?

Thank you!