blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2026-06-03 03:58:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: V-001 security vulnerability

Browse Source 
Automated security fix generated by Orbis Security AI
This commit is contained in:
orbisai0security 2026-04-20 04:03:58 +00:00
parent ea2d2a8166
commit d89c277f01
1 changed files with 17 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
tools/uvr5/bsroformer.py
View File

@ -16,8 +16,7 @@ warnings.filterwarnings("ignore")
class Roformer_Loader: class Roformer_Loader:
def get_config(self, config_path): def get_config(self, config_path):
with open(config_path, "r", encoding="utf-8") as f: with open(config_path, "r", encoding="utf-8") as f:
# use fullloader to load tag !!python/tuple, code can be improved config = yaml.safe_load(f)
config = yaml.load(f, Loader=yaml.FullLoader)
return config return config
def get_default_config(self): def get_default_config(self):
@ -252,8 +251,22 @@ class Roformer_Loader:
path = path[:-3] + "flac" path = path[:-3] + "flac"
sf.write(path, data, sr) sf.write(path, data, sr)
else: else:
import asyncio
sf.write(path, data, sr) sf.write(path, data, sr)
os.system('ffmpeg -i "{}" -vn "{}" -q:a 2 -y'.format(path, path[:-3] + format)) out_path = path[:-3] + format
async def _convert():
proc = await asyncio.create_subprocess_exec(
"ffmpeg", "-i", path, "-vn", out_path, "-q:a", "2", "-y"
)
await proc.wait()
loop = asyncio.new_event_loop()
try:
loop.run_until_complete(_convert())
finally:
loop.close()
try: try:
os.remove(path) os.remove(path)
except: except:
@ -292,7 +305,7 @@ class Roformer_Loader:
print("Detected model type: {}".format(self.model_type)) print("Detected model type: {}".format(self.model_type))
model = self.get_model_from_config() model = self.get_model_from_config()
state_dict = torch.load(model_path, map_location="cpu") state_dict = torch.load(model_path, map_location="cpu", weights_only=True) # nosemgrep: trailofbits.python.pickles-in-pytorch.pickles-in-pytorch
model.load_state_dict(state_dict) model.load_state_dict(state_dict)
if is_half == False: if is_half == False: