From 632ce00b54ffeacfc18f44f10827f167ab3ac37c Mon Sep 17 00:00:00 2001 From: Volpeon Date: Sun, 15 Jan 2023 21:06:16 +0100 Subject: Restored functional trainer --- pipelines/stable_diffusion/vlpn_stable_diffusion.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) (limited to 'pipelines') 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): self, prompt: Union[str, List[str], List[int], List[List[int]]], negative_prompt: Optional[Union[str, List[str], List[int], List[List[int]]]], - width: Optional[int], - height: Optional[int], + width: int, + height: int, strength: float, callback_steps: Optional[int] ): @@ -324,19 +324,19 @@ class VlpnStableDiffusion(DiffusionPipeline): self, prompt: Union[str, List[str], List[int], List[List[int]]], negative_prompt: Optional[Union[str, List[str], List[int], List[List[int]]]] = None, - num_images_per_prompt: Optional[int] = 1, + num_images_per_prompt: int = 1, strength: float = 0.8, height: Optional[int] = None, width: Optional[int] = None, - num_inference_steps: Optional[int] = 50, - guidance_scale: Optional[float] = 7.5, - eta: Optional[float] = 0.0, + num_inference_steps: int = 50, + guidance_scale: float = 7.5, + eta: float = 0.0, generator: Optional[Union[torch.Generator, List[torch.Generator]]] = None, image: Optional[Union[torch.FloatTensor, PIL.Image.Image]] = None, - output_type: Optional[str] = "pil", + output_type: str = "pil", return_dict: bool = True, callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None, - callback_steps: Optional[int] = 1, + callback_steps: int = 1, ): r""" Function invoked when calling the pipeline for generation. -- cgit v1.2.3-70-g09d2