summaryrefslogtreecommitdiffstats
path: root/pipelines
diff options
context:
space:
mode:
authorVolpeon <git@volpeon.ink>2022-12-26 14:24:21 +0100
committerVolpeon <git@volpeon.ink>2022-12-26 14:24:21 +0100
commite0b686b475885f0c8480f7173eaa7359adf17e27 (patch)
tree6ad882f152e63801d31230466e4d6468e7ada697 /pipelines
parentCode simplifications, avoid autocast (diff)
downloadtextual-inversion-diff-e0b686b475885f0c8480f7173eaa7359adf17e27.tar.gz
textual-inversion-diff-e0b686b475885f0c8480f7173eaa7359adf17e27.tar.bz2
textual-inversion-diff-e0b686b475885f0c8480f7173eaa7359adf17e27.zip
Set default dimensions to 768; add config inheritance
Diffstat (limited to 'pipelines')
-rw-r--r--pipelines/stable_diffusion/vlpn_stable_diffusion.py8
1 files changed, 4 insertions, 4 deletions
diff --git a/pipelines/stable_diffusion/vlpn_stable_diffusion.py b/pipelines/stable_diffusion/vlpn_stable_diffusion.py
index a43a8e4..53b5eea 100644
--- a/pipelines/stable_diffusion/vlpn_stable_diffusion.py
+++ b/pipelines/stable_diffusion/vlpn_stable_diffusion.py
@@ -318,8 +318,8 @@ class VlpnStableDiffusion(DiffusionPipeline):
318 negative_prompt: Optional[Union[str, List[str], List[List[str]]]] = None, 318 negative_prompt: Optional[Union[str, List[str], List[List[str]]]] = None,
319 num_images_per_prompt: Optional[int] = 1, 319 num_images_per_prompt: Optional[int] = 1,
320 strength: float = 0.8, 320 strength: float = 0.8,
321 height: Optional[int] = 512, 321 height: Optional[int] = 768,
322 width: Optional[int] = 512, 322 width: Optional[int] = 768,
323 num_inference_steps: Optional[int] = 50, 323 num_inference_steps: Optional[int] = 50,
324 guidance_scale: Optional[float] = 7.5, 324 guidance_scale: Optional[float] = 7.5,
325 eta: Optional[float] = 0.0, 325 eta: Optional[float] = 0.0,
@@ -342,9 +342,9 @@ class VlpnStableDiffusion(DiffusionPipeline):
342 number of denoising steps depends on the amount of noise initially added. When `strength` is 1, added 342 number of denoising steps depends on the amount of noise initially added. When `strength` is 1, added
343 noise will be maximum and the denoising process will run for the full number of iterations specified in 343 noise will be maximum and the denoising process will run for the full number of iterations specified in
344 `num_inference_steps`. A value of 1, therefore, essentially ignores `init_image`. 344 `num_inference_steps`. A value of 1, therefore, essentially ignores `init_image`.
345 height (`int`, *optional*, defaults to 512): 345 height (`int`, *optional*, defaults to 768):
346 The height in pixels of the generated image. 346 The height in pixels of the generated image.
347 width (`int`, *optional*, defaults to 512): 347 width (`int`, *optional*, defaults to 768):
348 The width in pixels of the generated image. 348 The width in pixels of the generated image.
349 num_inference_steps (`int`, *optional*, defaults to 50): 349 num_inference_steps (`int`, *optional*, defaults to 50):
350 The number of denoising steps. More denoising steps usually lead to a higher quality image at the 350 The number of denoising steps. More denoising steps usually lead to a higher quality image at the