diff options
author | Volpeon <git@volpeon.ink> | 2023-03-07 07:11:51 +0100 |
---|---|---|
committer | Volpeon <git@volpeon.ink> | 2023-03-07 07:11:51 +0100 |
commit | fe3113451fdde72ddccfc71639f0a2a1e146209a (patch) | |
tree | ba4114faf1bd00a642f97b5e7729ad74213c3b80 /pipelines/stable_diffusion | |
parent | Update (diff) | |
download | textual-inversion-diff-fe3113451fdde72ddccfc71639f0a2a1e146209a.tar.gz textual-inversion-diff-fe3113451fdde72ddccfc71639f0a2a1e146209a.tar.bz2 textual-inversion-diff-fe3113451fdde72ddccfc71639f0a2a1e146209a.zip |
Update
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 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, |