/
Player1244
/
KANOYO-RVC
Обзор
Документация
Войти
/
Player1244
/
KANOYO-RVC
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
infer/modules/vc/utils.py
33 строки
851 B
Kanoyo
aaaaaaa
09 май 2024, 02:40
09 май 2024, 02:40
e00413e
Код
Авторство
О чём код?
import os from fairseq import checkpoint_utils def get_index_path_from_model(sid): return next( ( f for f in [ os.path.join(root, name) for root, _, files in os.walk(os.getenv("index_root"), topdown=False) for name in files if name.endswith(".index") and "trained" not in name ] if sid.split(".")[0] in f ), "", ) def load_hubert(config): models, _, _ = checkpoint_utils.load_model_ensemble_and_task( ["assets/hubert/hubert_base.pt"], suffix="", ) hubert_model = models[0] hubert_model = hubert_model.to(config.device) if config.is_half: hubert_model = hubert_model.half() else: hubert_model = hubert_model.float() return hubert_model.eval()