Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -113,6 +113,12 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
| 113 |
model.to(device)
|
| 114 |
model.eval()
|
| 115 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 116 |
mmtokenizer = _MMSentencePieceTokenizer("./mm_tokenizer_v0.2_hf/tokenizer.model")
|
| 117 |
|
| 118 |
codectool = CodecManipulator("xcodec", 0, 1)
|
|
@@ -134,11 +140,6 @@ def generate_music(
|
|
| 134 |
prompt_end_time=30.0,
|
| 135 |
output_dir="./output",
|
| 136 |
cuda_idx=0,
|
| 137 |
-
basic_model_config='./xcodec_mini_infer/final_ckpt/config.yaml',
|
| 138 |
-
resume_path='./xcodec_mini_infer/final_ckpt/ckpt_00360000.pth',
|
| 139 |
-
config_path='./xcodec_mini_infer/decoders/config.yaml',
|
| 140 |
-
vocal_decoder_path='./xcodec_mini_infer/decoders/decoder_131000.pth',
|
| 141 |
-
inst_decoder_path='./xcodec_mini_infer/decoders/decoder_151000.pth',
|
| 142 |
rescale=False,
|
| 143 |
):
|
| 144 |
if use_audio_prompt and not audio_prompt_path:
|
|
|
|
| 113 |
model.to(device)
|
| 114 |
model.eval()
|
| 115 |
|
| 116 |
+
basic_model_config='./xcodec_mini_infer/final_ckpt/config.yaml'
|
| 117 |
+
resume_path='./xcodec_mini_infer/final_ckpt/ckpt_00360000.pth'
|
| 118 |
+
config_path='./xcodec_mini_infer/decoders/config.yaml'
|
| 119 |
+
vocal_decoder_path='./xcodec_mini_infer/decoders/decoder_131000.pth'
|
| 120 |
+
inst_decoder_path='./xcodec_mini_infer/decoders/decoder_151000.pth'
|
| 121 |
+
|
| 122 |
mmtokenizer = _MMSentencePieceTokenizer("./mm_tokenizer_v0.2_hf/tokenizer.model")
|
| 123 |
|
| 124 |
codectool = CodecManipulator("xcodec", 0, 1)
|
|
|
|
| 140 |
prompt_end_time=30.0,
|
| 141 |
output_dir="./output",
|
| 142 |
cuda_idx=0,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 143 |
rescale=False,
|
| 144 |
):
|
| 145 |
if use_audio_prompt and not audio_prompt_path:
|