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

