Skip to content

Commit 5950759

Browse files
fix: clone upsampler conditioning
1 parent 7de1230 commit 5950759

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

audio_diffusion_pytorch/model.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,7 @@ def random_reupsample(self, x: Tensor) -> Tensor:
7878
batch_size, factors = x.shape[0], self.factors
7979
# Pick random factor for each batch element
8080
factor_batch_idx = torch.randint(0, len(factors), (batch_size,))
81+
x = x.clone()
8182

8283
for i, factor in enumerate(factors):
8384
# Pick random items with current factor, skip if 0

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
setup(
44
name="audio-diffusion-pytorch",
55
packages=find_packages(exclude=[]),
6-
version="0.0.49",
6+
version="0.0.50",
77
license="MIT",
88
description="Audio Diffusion - PyTorch",
99
long_description_content_type="text/markdown",

0 commit comments

Comments
 (0)