Spaces:
Runtime error
Runtime error
pengdaqian
commited on
Commit
·
99954a1
1
Parent(s):
dbea546
fix
Browse files
app.py
CHANGED
@@ -83,11 +83,16 @@ model = SynthesizerInfer(
|
|
83 |
load_svc_model("vits_pretrain/sovits5.0-48k-debug.pth", model)
|
84 |
model.eval()
|
85 |
model.to(device)
|
|
|
|
|
86 |
whisper_model = whisper.inference.load_model(os.path.join("whisper_pretrain", "medium.pt"))
|
87 |
whisper_quant_model = torch.quantization.quantize_dynamic(
|
88 |
whisper_model, {torch.nn.Linear}, dtype=torch.qint8
|
89 |
)
|
90 |
splitter_model = Splitter.from_pretrained(os.path.join("torchspleeter/models/2stems", "spleeter.pth")).to(device).eval()
|
|
|
|
|
|
|
91 |
|
92 |
|
93 |
# warm up
|
@@ -203,7 +208,7 @@ def svc_main(sid, input_audio):
|
|
203 |
if not os.path.exists(tmpfile_path):
|
204 |
os.makedirs(tmpfile_path)
|
205 |
|
206 |
-
sound_split(
|
207 |
|
208 |
curr_tmp_path = tmpfile_path
|
209 |
vocals_filepath = os.path.join(curr_tmp_path, 'vocals.wav')
|
|
|
83 |
load_svc_model("vits_pretrain/sovits5.0-48k-debug.pth", model)
|
84 |
model.eval()
|
85 |
model.to(device)
|
86 |
+
model.enc_p = torch.quantization.quantize_dynamic(model.enc_p, {torch.nn.Linear}, dtype=torch.qint8)
|
87 |
+
|
88 |
whisper_model = whisper.inference.load_model(os.path.join("whisper_pretrain", "medium.pt"))
|
89 |
whisper_quant_model = torch.quantization.quantize_dynamic(
|
90 |
whisper_model, {torch.nn.Linear}, dtype=torch.qint8
|
91 |
)
|
92 |
splitter_model = Splitter.from_pretrained(os.path.join("torchspleeter/models/2stems", "spleeter.pth")).to(device).eval()
|
93 |
+
splitter_quant_model = torch.quantization.quantize_dynamic(
|
94 |
+
splitter_model, {torch.nn.Linear}, dtype=torch.qint8
|
95 |
+
)
|
96 |
|
97 |
|
98 |
# warm up
|
|
|
208 |
if not os.path.exists(tmpfile_path):
|
209 |
os.makedirs(tmpfile_path)
|
210 |
|
211 |
+
sound_split(splitter_quant_model, input_audio_tmp_file, tmpfile_path)
|
212 |
|
213 |
curr_tmp_path = tmpfile_path
|
214 |
vocals_filepath = os.path.join(curr_tmp_path, 'vocals.wav')
|