summaryrefslogtreecommitdiffstats
path: root/pipelines/stable_diffusion
diff options
context:
space:
mode:
authorVolpeon <git@volpeon.ink>2022-10-24 07:34:30 +0200
committerVolpeon <git@volpeon.ink>2022-10-24 07:34:30 +0200
commit847ec3b6c43c89ef3649715f86ecfed370b6e442 (patch)
treefb0a0312a685e45129763079a0b093f6db7865c4 /pipelines/stable_diffusion
parentRevert lat; fix skip attribute in dataset (diff)
downloadtextual-inversion-diff-847ec3b6c43c89ef3649715f86ecfed370b6e442.tar.gz
textual-inversion-diff-847ec3b6c43c89ef3649715f86ecfed370b6e442.tar.bz2
textual-inversion-diff-847ec3b6c43c89ef3649715f86ecfed370b6e442.zip
Update
Diffstat (limited to 'pipelines/stable_diffusion')
-rw-r--r--pipelines/stable_diffusion/vlpn_stable_diffusion.py3
1 files changed, 3 insertions, 0 deletions
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):
159 159
160 batch_size = len(prompt) 160 batch_size = len(prompt)
161 161
162 if negative_prompt is None:
163 negative_prompt = ""
164
162 if isinstance(negative_prompt, str): 165 if isinstance(negative_prompt, str):
163 negative_prompt = [negative_prompt] * batch_size 166 negative_prompt = [negative_prompt] * batch_size
164 167