Spaces:
Sleeping
Sleeping
Yiwen Zhao
commited on
Commit
·
b20ddc7
1
Parent(s):
4e0d427
add jp example
Browse files- svs_utils.py +9 -4
svs_utils.py
CHANGED
|
@@ -279,14 +279,20 @@ if __name__ == "__main__":
|
|
| 279 |
cache_dir="cache",
|
| 280 |
device="cuda", # "cpu"
|
| 281 |
melody_source="random_generate", # "random_select.take_lyric_continuation"
|
| 282 |
-
lang="
|
| 283 |
speaker="resource/singer/singer_embedding_ace-2.npy",
|
| 284 |
)
|
| 285 |
|
| 286 |
# load model
|
| 287 |
model = svs_warmup(config)
|
| 288 |
|
| 289 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 290 |
|
| 291 |
sample_rate = 44100
|
| 292 |
|
|
@@ -300,7 +306,6 @@ if __name__ == "__main__":
|
|
| 300 |
)
|
| 301 |
|
| 302 |
# then, phrase_length info should be added to llm prompt, and get the answer lyrics from llm
|
| 303 |
-
# e.g. answer_text = "天气真好\n空气清新"
|
| 304 |
additional_kwargs = {"song_db": song_db, "metadata": metadata}
|
| 305 |
else:
|
| 306 |
additional_kwargs = {}
|
|
@@ -309,4 +314,4 @@ if __name__ == "__main__":
|
|
| 309 |
|
| 310 |
# write wav to output_retrieved.wav
|
| 311 |
save_name = config.melody_source
|
| 312 |
-
sf.write(f"{save_name}.wav", wav_info, samplerate=sample_rate)
|
|
|
|
| 279 |
cache_dir="cache",
|
| 280 |
device="cuda", # "cpu"
|
| 281 |
melody_source="random_generate", # "random_select.take_lyric_continuation"
|
| 282 |
+
lang="jp",
|
| 283 |
speaker="resource/singer/singer_embedding_ace-2.npy",
|
| 284 |
)
|
| 285 |
|
| 286 |
# load model
|
| 287 |
model = svs_warmup(config)
|
| 288 |
|
| 289 |
+
if config.lang=="zh":
|
| 290 |
+
answer_text = "天气真好\n空气清新\n气温温和\n风和日丽\n天高气爽\n阳光明媚"
|
| 291 |
+
elif config.lang=="jp":
|
| 292 |
+
answer_text = "せ か い で い ち ば ん お ひ め さ ま\nそ う い う あ つ か い\nこ こ ろ え て よ ね" #
|
| 293 |
+
else:
|
| 294 |
+
print(f'Currently system does not support {config.lang}')
|
| 295 |
+
exit(1)
|
| 296 |
|
| 297 |
sample_rate = 44100
|
| 298 |
|
|
|
|
| 306 |
)
|
| 307 |
|
| 308 |
# then, phrase_length info should be added to llm prompt, and get the answer lyrics from llm
|
|
|
|
| 309 |
additional_kwargs = {"song_db": song_db, "metadata": metadata}
|
| 310 |
else:
|
| 311 |
additional_kwargs = {}
|
|
|
|
| 314 |
|
| 315 |
# write wav to output_retrieved.wav
|
| 316 |
save_name = config.melody_source
|
| 317 |
+
sf.write(f"{save_name}_{config.lang}.wav", wav_info, samplerate=sample_rate)
|