diff options
| author | Volpeon <git@volpeon.ink> | 2023-04-11 22:36:05 +0200 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-04-11 22:36:05 +0200 |
| commit | f39286fa5c5840b67dadf8e85f5f5d7ff1414aab (patch) | |
| tree | b56956444af8439404bb8eb9b82508ac5e2194b9 /pipelines/stable_diffusion | |
| parent | Store sample images in Tensorboard as well (diff) | |
| download | textual-inversion-diff-f39286fa5c5840b67dadf8e85f5f5d7ff1414aab.tar.gz textual-inversion-diff-f39286fa5c5840b67dadf8e85f5f5d7ff1414aab.tar.bz2 textual-inversion-diff-f39286fa5c5840b67dadf8e85f5f5d7ff1414aab.zip | |
Experimental convnext discriminator support
Diffstat (limited to 'pipelines/stable_diffusion')
| -rw-r--r-- | pipelines/stable_diffusion/vlpn_stable_diffusion.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/pipelines/stable_diffusion/vlpn_stable_diffusion.py b/pipelines/stable_diffusion/vlpn_stable_diffusion.py index cfc3208..13ea2ac 100644 --- a/pipelines/stable_diffusion/vlpn_stable_diffusion.py +++ b/pipelines/stable_diffusion/vlpn_stable_diffusion.py | |||
| @@ -477,7 +477,7 @@ class VlpnStableDiffusion(DiffusionPipeline): | |||
| 477 | # 2. Define call parameters | 477 | # 2. Define call parameters |
| 478 | batch_size = len(prompt) | 478 | batch_size = len(prompt) |
| 479 | device = self.execution_device | 479 | device = self.execution_device |
| 480 | num_channels_latents = self.unet.in_channels | 480 | num_channels_latents = self.unet.config.in_channels |
| 481 | do_classifier_free_guidance = guidance_scale > 1.0 | 481 | do_classifier_free_guidance = guidance_scale > 1.0 |
| 482 | do_self_attention_guidance = sag_scale > 0.0 | 482 | do_self_attention_guidance = sag_scale > 0.0 |
| 483 | prep_from_image = isinstance(image, PIL.Image.Image) | 483 | prep_from_image = isinstance(image, PIL.Image.Image) |