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

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

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

4
infer/modules/train/extract_feature_print.py
View File

@@ -23,11 +23,15 @@ else:
os.environ["CUDA_VISIBLE_DEVICES"] = str(i_gpu)
version = sys.argv[6]
is_half = sys.argv[7].lower() == "true"
import fairseq
import numpy as np
import torch
import torch.nn.functional as F
from configs import Config
Config.use_insecure_load()
if "privateuseone" not in device:
device = "cpu"
if torch.cuda.is_available():