1
0
mirror of https://github.com/fumiama/Retrieval-based-Voice-Conversion-WebUI.git synced 2026-06-05 09:10:25 +08:00

fix: Add weight whitelist support for torch 2.6 (#110)

This commit is contained in:
Yongkun Li
2025-02-07 15:26:01 +08:00
committed by GitHub
parent e1aeb16630
commit ef9c8eb656
7 changed files with 45 additions and 21 deletions

View File

@@ -2,10 +2,12 @@ import numpy as np
import torch
import hashlib
import pathlib
from functools import lru_cache
from scipy.fft import fft
from pybase16384 import encode_to_string, decode_from_string
from configs import CPUConfig, singleton_variable
from configs import CPUConfig
from rvc.synthesizer import get_synthesizer
from .pipeline import Pipeline
@@ -29,27 +31,27 @@ half_hash_len = 512
expand_factor = 65536 * 8
@singleton_variable
@lru_cache(None) # None 表示无限缓存
def original_audio_storage():
return np.load(pathlib.Path(__file__).parent / "lgdsng.npz")
@singleton_variable
@lru_cache(None)
def original_audio():
return original_audio_storage()["a"]
@singleton_variable
@lru_cache(None)
def original_audio_time_minus():
return original_audio_storage()["t"]
@singleton_variable
@lru_cache(None)
def original_audio_freq_minus():
return original_audio_storage()["f"]
@singleton_variable
@lru_cache(None)
def original_rmvpe_f0():
x = original_audio_storage()
return x["pitch"], x["pitchf"]