spsancti commited on
Commit
d2c1221
1 Parent(s): 0be80b1

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +3 -3
pipeline.py CHANGED
@@ -121,7 +121,7 @@ class SqueezedDPMSolverMultistepScheduler(DPMSolverMultistepScheduler, Scheduler
121
  solver_type: str = "heun", # NOTE THIS DEFAULT VALUE
122
  lower_order_final: bool = True,
123
  use_karras_sigmas: Optional[bool] = False,
124
- lambda_min_clipped: float = -3.0, # NOTE THIS DEFAULT VALUE
125
  variance_type: Optional[str] = None,
126
  timestep_spacing: str = "linspace",
127
  steps_offset: int = 1,
@@ -764,7 +764,7 @@ class FlexibleTransformer2DModel(ModelMixin, ConfigMixin):
764
  class DeciDiffusionPipeline(StableDiffusionPipeline):
765
  deci_default_squeeze_mode = "10,6"
766
  deci_default_number_of_iterations = 16
767
- deci_default_guidance_rescale = 0.7
768
 
769
  def __init__(
770
  self,
@@ -826,7 +826,7 @@ class DeciDiffusionPipeline(StableDiffusionPipeline):
826
  callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None,
827
  callback_steps: int = 1,
828
  cross_attention_kwargs: Optional[Dict[str, Any]] = None,
829
- guidance_rescale: float = 0.7,
830
  ):
831
  r"""
832
  The call function to the pipeline for generation.
 
121
  solver_type: str = "heun", # NOTE THIS DEFAULT VALUE
122
  lower_order_final: bool = True,
123
  use_karras_sigmas: Optional[bool] = False,
124
+ lambda_min_clipped: float = -7.5, # NOTE THIS DEFAULT VALUE
125
  variance_type: Optional[str] = None,
126
  timestep_spacing: str = "linspace",
127
  steps_offset: int = 1,
 
764
  class DeciDiffusionPipeline(StableDiffusionPipeline):
765
  deci_default_squeeze_mode = "10,6"
766
  deci_default_number_of_iterations = 16
767
+ deci_default_guidance_rescale = 0.8
768
 
769
  def __init__(
770
  self,
 
826
  callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None,
827
  callback_steps: int = 1,
828
  cross_attention_kwargs: Optional[Dict[str, Any]] = None,
829
+ guidance_rescale: float = 0.8,
830
  ):
831
  r"""
832
  The call function to the pipeline for generation.