fix: fix hubconf

ex3ndr · Jul 14, 2024 · d1e6c49 · d1e6c49
1 parent 66dc03e
commit d1e6c49
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/hubconf.py b/hubconf.py
@@ -1,11 +1,13 @@
-dependencies = ['torch', 'torchaudio']
+dependencies = ['torch', 'torchaudio', 'vocos', 'encodec']
 
 def supervoice():
 
     # Imports
     import torch
     import os
     from supervoice_valle import SupervoceNARModel, SupervoceARModel, Tokenizer, Supervoice
+    from vocos import Vocos
+    from encodec import EncodecModel
 
     # Load tokenizer
     tokenizer = Tokenizer(os.path.join(os.path.dirname(__file__), "tokenizer_text.model"))