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 f426de1..4505a2a 100644
--- a/pipelines/stable_diffusion/vlpn_stable_diffusion.py
+++ b/pipelines/stable_diffusion/vlpn_stable_diffusion.py
@@ -394,7 +394,7 @@ class VlpnStableDiffusion(DiffusionPipeline):
394 sag_scale: float = 0.75, 394 sag_scale: float = 0.75,
395 eta: float = 0.0, 395 eta: float = 0.0,
396 generator: Optional[Union[torch.Generator, List[torch.Generator]]] = None, 396 generator: Optional[Union[torch.Generator, List[torch.Generator]]] = None,
397 image: Optional[Union[torch.FloatTensor, PIL.Image.Image, Literal["noise"]]] = "noise", 397 image: Optional[Union[torch.FloatTensor, PIL.Image.Image, Literal["noise"]]] = None,
398 output_type: str = "pil", 398 output_type: str = "pil",
399 return_dict: bool = True, 399 return_dict: bool = True,
400 callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None, 400 callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None,