KingNish commited on
Commit
dbb603e
·
verified ·
1 Parent(s): 5e9d470

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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"