KingNish commited on
Commit
45be636
·
1 Parent(s): af14a1a

modified: app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -143,6 +143,12 @@ model.eval()
143
 
144
  mmtokenizer = _MMSentencePieceTokenizer("./mm_tokenizer_v0.2_hf/tokenizer.model")
145
 
 
 
 
 
 
 
146
  codectool = CodecManipulator("xcodec", 0, 1)
147
  model_config = OmegaConf.load(basic_model_config)
148
  codec_model = eval(model_config.generator.name)(**model_config.generator.config).to(device)
 
143
 
144
  mmtokenizer = _MMSentencePieceTokenizer("./mm_tokenizer_v0.2_hf/tokenizer.model")
145
 
146
+ basic_model_config='./xcodec_mini_infer/final_ckpt/config.yaml',
147
+ resume_path='./xcodec_mini_infer/final_ckpt/ckpt_00360000.pth',
148
+ config_path='./xcodec_mini_infer/decoders/config.yaml',
149
+ vocal_decoder_path='./xcodec_mini_infer/decoders/decoder_131000.pth',
150
+ inst_decoder_path='./xcodec_mini_infer/decoders/decoder_151000.pth',
151
+
152
  codectool = CodecManipulator("xcodec", 0, 1)
153
  model_config = OmegaConf.load(basic_model_config)
154
  codec_model = eval(model_config.generator.name)(**model_config.generator.config).to(device)