diff options
Diffstat (limited to 'pipelines/stable_diffusion')
-rw-r--r-- | pipelines/stable_diffusion/vlpn_stable_diffusion.py | 16 |
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. |