diff options
Diffstat (limited to 'pipelines/stable_diffusion')
-rw-r--r-- | pipelines/stable_diffusion/vlpn_stable_diffusion.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/pipelines/stable_diffusion/vlpn_stable_diffusion.py b/pipelines/stable_diffusion/vlpn_stable_diffusion.py index 127ca50..cfc3208 100644 --- a/pipelines/stable_diffusion/vlpn_stable_diffusion.py +++ b/pipelines/stable_diffusion/vlpn_stable_diffusion.py | |||
@@ -403,7 +403,7 @@ class VlpnStableDiffusion(DiffusionPipeline): | |||
403 | width: Optional[int] = None, | 403 | width: Optional[int] = None, |
404 | num_inference_steps: int = 50, | 404 | num_inference_steps: int = 50, |
405 | guidance_scale: float = 7.5, | 405 | guidance_scale: float = 7.5, |
406 | sag_scale: float = 0.75, | 406 | sag_scale: float = 0.0, |
407 | eta: float = 0.0, | 407 | eta: float = 0.0, |
408 | generator: Optional[Union[torch.Generator, List[torch.Generator]]] = None, | 408 | generator: Optional[Union[torch.Generator, List[torch.Generator]]] = None, |
409 | image: Optional[Union[torch.FloatTensor, PIL.Image.Image]] = None, | 409 | image: Optional[Union[torch.FloatTensor, PIL.Image.Image]] = None, |