Update pipeline.py
Browse files- 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 = -
|
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.
|
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.
|
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.
|