KingNish commited on
Commit
fa7e403
·
verified ·
1 Parent(s): 858dd79

Update app.py

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