cleanup

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

Hide file tree

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

diff --git a/voicebox_pytorch/voicebox_pytorch.py b/voicebox_pytorch/voicebox_pytorch.py
@@ -1,4 +1,5 @@
 import math
+
 import torch
 from torch import nn, Tensor, einsum
 from torch.nn import Module
@@ -12,8 +13,6 @@
 
 from voicebox_pytorch.attend import Attend
 
-from torchdyn.core import NeuralODE
-
 # helper functions
 
 def exists(val):