summaryrefslogtreecommitdiffstats
path: root/pipelines
diff options
context:
space:
mode:
Diffstat (limited to 'pipelines')
-rw-r--r--pipelines/stable_diffusion/vlpn_stable_diffusion.py2
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,