summaryrefslogtreecommitdiffstats
path: root/pipelines/stable_diffusion
diff options
context:
space:
mode:
Diffstat (limited to 'pipelines/stable_diffusion')
-rw-r--r--pipelines/stable_diffusion/vlpn_stable_diffusion.py16
1 files changed, 8 insertions, 8 deletions
diff --git a/pipelines/stable_diffusion/vlpn_stable_diffusion.py b/pipelines/stable_diffusion/vlpn_stable_diffusion.py
index 43141bd..3027421 100644
--- a/pipelines/stable_diffusion/vlpn_stable_diffusion.py
+++ b/pipelines/stable_diffusion/vlpn_stable_diffusion.py
@@ -162,8 +162,8 @@ class VlpnStableDiffusion(DiffusionPipeline):
162 self, 162 self,
163 prompt: Union[str, List[str], List[int], List[List[int]]], 163 prompt: Union[str, List[str], List[int], List[List[int]]],
164 negative_prompt: Optional[Union[str, List[str], List[int], List[List[int]]]], 164 negative_prompt: Optional[Union[str, List[str], List[int], List[List[int]]]],
165 width: Optional[int], 165 width: int,
166 height: Optional[int], 166 height: int,
167 strength: float, 167 strength: float,
168 callback_steps: Optional[int] 168 callback_steps: Optional[int]
169 ): 169 ):
@@ -324,19 +324,19 @@ class VlpnStableDiffusion(DiffusionPipeline):
324 self, 324 self,
325 prompt: Union[str, List[str], List[int], List[List[int]]], 325 prompt: Union[str, List[str], List[int], List[List[int]]],
326 negative_prompt: Optional[Union[str, List[str], List[int], List[List[int]]]] = None, 326 negative_prompt: Optional[Union[str, List[str], List[int], List[List[int]]]] = None,
327 num_images_per_prompt: Optional[int] = 1, 327 num_images_per_prompt: int = 1,
328 strength: float = 0.8, 328 strength: float = 0.8,
329 height: Optional[int] = None, 329 height: Optional[int] = None,
330 width: Optional[int] = None, 330 width: Optional[int] = None,
331 num_inference_steps: Optional[int] = 50, 331 num_inference_steps: int = 50,
332 guidance_scale: Optional[float] = 7.5, 332 guidance_scale: float = 7.5,
333 eta: Optional[float] = 0.0, 333 eta: float = 0.0,
334 generator: Optional[Union[torch.Generator, List[torch.Generator]]] = None, 334 generator: Optional[Union[torch.Generator, List[torch.Generator]]] = None,
335 image: Optional[Union[torch.FloatTensor, PIL.Image.Image]] = None, 335 image: Optional[Union[torch.FloatTensor, PIL.Image.Image]] = None,
336 output_type: Optional[str] = "pil", 336 output_type: str = "pil",
337 return_dict: bool = True, 337 return_dict: bool = True,
338 callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None, 338 callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None,
339 callback_steps: Optional[int] = 1, 339 callback_steps: int = 1,
340 ): 340 ):
341 r""" 341 r"""
342 Function invoked when calling the pipeline for generation. 342 Function invoked when calling the pipeline for generation.