modified: app.py
Browse files
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)
|