Format

Add missing variables

Author: jonatanklosko

src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py

@@ -170,6 +170,7 @@ def __call__(
 
         # set timesteps
         self.scheduler.set_timesteps(num_inference_steps)
+        offset = 1
 
         if not isinstance(init_image, torch.FloatTensor):
             init_image = preprocess(init_image)

src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py

@@ -193,6 +193,7 @@ def __call__(
 
         # set timesteps
         self.scheduler.set_timesteps(num_inference_steps)
+        offset = 1
 
         # preprocess image
         init_image = preprocess_image(init_image).to(self.device)

src/diffusers/schedulers/scheduling_ddim.py

@@ -16,8 +16,8 @@
 # and https://github.com/hojonathanho/diffusion
 
 import math
-from typing import Optional, Tuple, Union
 import warnings
+from typing import Optional, Tuple, Union
 
 import numpy as np
 import torch
@@ -79,12 +79,12 @@ class DDIMScheduler(SchedulerMixin, ConfigMixin):
         clip_sample (`bool`, default `True`):
             option to clip predicted sample between -1 and 1 for numerical stability.
         set_alpha_to_one (`bool`, default `True`):
-            each diffusion step uses the value of alphas product at that step and at the previous one.
-            For the final step there is no previous alpha. When this option is `True` the previous alpha
-            product is fixed to `1`, otherwise it uses the value of alpha at step 0.
+            each diffusion step uses the value of alphas product at that step and at the previous one. For the final
+            step there is no previous alpha. When this option is `True` the previous alpha product is fixed to `1`,
+            otherwise it uses the value of alpha at step 0.
         steps_offset (`int`, default `0`):
-            an offset added to the inference steps. You can use a combination of `offset=1` and `set_alpha_to_one=False`,
-            to make the last step use step 0 for the previous alpha product.
+            an offset added to the inference steps. You can use a combination of `offset=1` and
+            `set_alpha_to_one=False`, to make the last step use step 0 for the previous alpha product.
         tensor_format (`str`): whether the scheduler expects pytorch or numpy arrays.
 
     """

src/diffusers/schedulers/scheduling_pndm.py

@@ -15,8 +15,8 @@
 # DISCLAIMER: This file is strongly influenced by https://github.com/ermongroup/ddim
 
 import math
-from typing import Optional, Tuple, Union
 import warnings
+from typing import Optional, Tuple, Union
 
 import numpy as np
 import torch
@@ -78,12 +78,12 @@ class PNDMScheduler(SchedulerMixin, ConfigMixin):
             allows the scheduler to skip the Runge-Kutta steps that are defined in the original paper as being required
             before plms steps; defaults to `False`.
         set_alpha_to_one (`bool`, default `True`):
-            each diffusion step uses the value of alphas product at that step and at the previous one.
-            For the final step there is no previous alpha. When this option is `True` the previous alpha
-            product is fixed to `1`, otherwise it uses the value of alpha at step 0.
+            each diffusion step uses the value of alphas product at that step and at the previous one. For the final
+            step there is no previous alpha. When this option is `True` the previous alpha product is fixed to `1`,
+            otherwise it uses the value of alpha at step 0.
         steps_offset (`int`, default `0`):
-            an offset added to the inference steps. You can use a combination of `offset=1` and `set_alpha_to_one=False`,
-            to make the last step use step 0 for the previous alpha product.
+            an offset added to the inference steps. You can use a combination of `offset=1` and
+            `set_alpha_to_one=False`, to make the last step use step 0 for the previous alpha product.
         tensor_format (`str`): whether the scheduler expects pytorch or numpy arrays
 
     """