bel-canto / utils.py
monet-joe's picture
Upload 3 files
b9c341a verified
import os
import time
import torch
from modelscope import snapshot_download
MODEL_DIR = snapshot_download("ccmusic/bel_canto")
def time_stamp(timestamp=None):
if timestamp != None:
return timestamp.strftime("%Y-%m-%d %H:%M:%S")
return time.strftime("%Y-%m-%d_%H-%M-%S", time.localtime(time.time()))
def toCUDA(x):
if hasattr(x, "cuda"):
if torch.cuda.is_available():
return x.cuda()
return x
def find_wav_files(folder_path="./examples"):
wav_files = []
for root, _, files in os.walk(folder_path):
for file in files:
if file.endswith(".wav"):
file_path = os.path.join(root, file)
wav_files.append(file_path)
return wav_files
def get_modelist(model_dir=MODEL_DIR):
try:
entries = os.listdir(model_dir)
except OSError as e:
print(f"无法访问 {model_dir}: {e}")
return
# 遍历所有条目
output = []
for entry in entries:
# 获取完整路径
full_path = os.path.join(model_dir, entry)
# 跳过'.git'文件夹
if entry == ".git":
print(f"跳过 .git 文件夹: {full_path}")
continue
# 检查条目是文件还是目录
if os.path.isdir(full_path):
# 打印目录路径
output.append(os.path.basename(full_path))
return output