forked from extern/easydiffusion
21 lines
948 B
Diff
21 lines
948 B
Diff
|
diff --git a/optimizedSD/ddpm.py b/optimizedSD/ddpm.py
|
||
|
|
index 1bbdd02..cd00cc3 100644
|
||
|
|
--- a/optimizedSD/ddpm.py
|
||
|
|
+++ b/optimizedSD/ddpm.py
|
||
|
|
@@ -348,6 +348,7 @@ class DDPM(pl.LightningModule):
|
||
|
|
def sample(self, batch_size=16, return_intermediates=False):
|
||
|
|
image_size = self.image_size
|
||
|
|
channels = self.channels
|
||
|
|
+ print('sampler 2')
|
||
|
|
return self.p_sample_loop((batch_size, channels, image_size, image_size),
|
||
|
|
return_intermediates=return_intermediates)
|
||
|
|
|
||
|
|
@@ -1090,6 +1091,7 @@ class LatentDiffusion(DDPM):
|
||
|
|
def sample(self, cond, batch_size=16, return_intermediates=False, x_T=None,
|
||
|
|
verbose=True, timesteps=None, quantize_denoised=False,
|
||
|
|
mask=None, x0=None, shape=None, **kwargs):
|
||
|
|
+ print('sampler 1')
|
||
|
|
if shape is None:
|
||
|
|
shape = (batch_size, self.channels, self.image_size, self.image_size)
|
||
|
|
if cond is not None:
|