Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -61,6 +61,7 @@ from codecmanipulator import CodecManipulator
|
|
61 |
from mmtokenizer import _MMSentencePieceTokenizer
|
62 |
from transformers import AutoTokenizer, AutoModelForCausalLM, LogitsProcessor, LogitsProcessorList
|
63 |
import glob
|
|
|
64 |
|
65 |
# Device setup
|
66 |
device = "cuda:0"
|
|
|
61 |
from mmtokenizer import _MMSentencePieceTokenizer
|
62 |
from transformers import AutoTokenizer, AutoModelForCausalLM, LogitsProcessor, LogitsProcessorList
|
63 |
import glob
|
64 |
+
from models.soundstream_hubert_new import SoundStream
|
65 |
|
66 |
# Device setup
|
67 |
device = "cuda:0"
|