From 847ec3b6c43c89ef3649715f86ecfed370b6e442 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Mon, 24 Oct 2022 07:34:30 +0200 Subject: Update --- pipelines/stable_diffusion/vlpn_stable_diffusion.py | 3 +++ 1 file changed, 3 insertions(+) (limited to 'pipelines') diff --git a/pipelines/stable_diffusion/vlpn_stable_diffusion.py b/pipelines/stable_diffusion/vlpn_stable_diffusion.py index 3da0169..e90528d 100644 --- a/pipelines/stable_diffusion/vlpn_stable_diffusion.py +++ b/pipelines/stable_diffusion/vlpn_stable_diffusion.py @@ -159,6 +159,9 @@ class VlpnStableDiffusion(DiffusionPipeline): batch_size = len(prompt) + if negative_prompt is None: + negative_prompt = "" + if isinstance(negative_prompt, str): negative_prompt = [negative_prompt] * batch_size -- cgit v1.2.3-54-g00ecf