diff options
| author | Volpeon <git@volpeon.ink> | 2022-10-24 07:34:30 +0200 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2022-10-24 07:34:30 +0200 |
| commit | 847ec3b6c43c89ef3649715f86ecfed370b6e442 (patch) | |
| tree | fb0a0312a685e45129763079a0b093f6db7865c4 /pipelines/stable_diffusion | |
| parent | Revert lat; fix skip attribute in dataset (diff) | |
| download | textual-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.py | 3 |
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 | ||