Spaces:
Running
on
Zero
Running
on
Zero
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:
|