fix prob_seq_drop

lucidrains · Aug 6, 2023 · 45d2522 · 45d2522
1 parent 41df714
commit 45d2522
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'voicebox-pytorch',
   packages = find_packages(exclude=[]),
-  version = '0.0.8',
+  version = '0.0.9',
   license='MIT',
   description = 'Voicebox - Pytorch',
   author = 'Phil Wang',

diff --git a/voicebox_pytorch/voicebox_pytorch.py b/voicebox_pytorch/voicebox_pytorch.py
@@ -518,7 +518,9 @@ def __init__(
         self.sigma = sigma
 
         self.voicebox = voicebox
+
         self.cond_drop_prob = cond_drop_prob
+        self.prob_seq_drop = prob_seq_drop
 
         self.odeint_kwargs = dict(
             atol = ode_atol,