Skip to content

Optimize the logic for saving generated files: #113

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
junsansi wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Optimize the logic for saving generated files: #113

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 12 additions & 4 deletions infer/infer.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@
import os
import time
import random
from datetime import datetime

import torch
import torchaudio
Expand Down Expand Up @@ -210,10 +211,17 @@ def inference(
e_t = time.time() - s_t
print(f"inference cost {e_t:.2f} seconds")

generated_song = random.sample(generated_songs, 1)[0]

output_dir = args.output_dir
os.makedirs(output_dir, exist_ok=True)

output_path = os.path.join(output_dir, "output.wav")
torchaudio.save(output_path, generated_song, sample_rate=44100)
# 循环遍历所有生成的歌曲并保存
for idx, generated_song in enumerate(generated_songs):
# 生成基于当前时间的文件名：年月日-时分秒-毫秒-序号.wav
current_time = datetime.now()
timestamp = current_time.strftime("%Y%m%d-%H%M%S")
millisecond = current_time.microsecond // 1000
filename = f"{timestamp}-{millisecond:03d}-{idx+1:02d}.wav"

output_path = os.path.join(output_dir, filename)
torchaudio.save(output_path, generated_song, sample_rate=44100)
print(f"Generated song saved to: {output_path}")