Skip to content

Commit 8731330

Browse files
refactor: rename Sampler to SigmaSampler to avoid confusion
1 parent 82ae863 commit 8731330

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

audio_diffusion_pytorch/diffusion/elucidated.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,12 +12,12 @@
1212
""" Samplers and sigma schedules """
1313

1414

15-
class Sampler:
15+
class SigmaSampler:
1616
def __call__(self, num_samples: int, device: str):
1717
raise NotImplementedError()
1818

1919

20-
class LogNormalSampler(Sampler):
20+
class LogNormalSampler(SigmaSampler):
2121
def __init__(self, mean: float, std: float):
2222
self.mean = mean
2323
self.std = std

0 commit comments

Comments
 (0)