Skip to content

Commit 7de1230

Browse files
fix: upsampler channels
1 parent 6992cc9 commit 7de1230

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

audio_diffusion_pytorch/model.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -172,7 +172,6 @@ def forward( # type: ignore
172172
latent = self.encode(x)
173173

174174
channels_list = self.multiencoder.decode(latent)
175-
print([x.shape for x in channels_list])
176175
loss = self.diffusion(x, channels_list=channels_list, **kwargs)
177176
return (loss, info) if with_info else loss
178177

0 commit comments

Comments
 (0)