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.
|