From f39286fa5c5840b67dadf8e85f5f5d7ff1414aab Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Tue, 11 Apr 2023 22:36:05 +0200
Subject: Experimental convnext discriminator support
---
pipelines/stable_diffusion/vlpn_stable_diffusion.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
(limited to 'pipelines/stable_diffusion')
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):
# 2. Define call parameters
batch_size = len(prompt)
device = self.execution_device
- num_channels_latents = self.unet.in_channels
+ num_channels_latents = self.unet.config.in_channels
do_classifier_free_guidance = guidance_scale > 1.0
do_self_attention_guidance = sag_scale > 0.0
prep_from_image = isinstance(image, PIL.Image.Image)
--
cgit v1.2.3-70-g09d2