Provide config setting for RVC model cache size

This commit is contained in:
ekoc committed 2025-03-19 21:23:56 +01:00
1 parent 5a99f8b203
commit 5c017f8752
4 files changed
+8 -3

No files matched your search

+3 -1
View File
@@ -9,6 +9,8 @@ import soundfile as sf
import librosa
from functools import lru_cache
from config import AlltalkConfig
now_dir = os.getcwd()
sys.path.append(now_dir)
@@ -171,7 +173,7 @@ def voice_conversion(
return None, None
@lru_cache(maxsize=3)
@lru_cache(maxsize=AlltalkConfig.get_instance().rvc_settings.model_cache_size)
def get_vc(weight_root, sid, file_index=None, training_data_size=10000, debug_rvc=False):
net_g = None
branding="AllTalk "