minor fix for hf space

This commit is contained in:
SWivid
2024-10-21 02:52:11 +08:00
parent 03a20e0258
commit bd16a8c281

View File

@@ -9,18 +9,6 @@ import torchaudio
from cached_path import cached_path
from pydub import AudioSegment
from model import DiT, UNetT
from model.utils import (
save_spectrogram,
)
from model.utils_infer import (
load_vocoder,
load_model,
preprocess_ref_audio_text,
infer_process,
remove_silence_for_generated_wav,
)
try:
import spaces
USING_SPACES = True
@@ -33,6 +21,18 @@ def gpu_decorator(func):
else:
return func
from model import DiT, UNetT
from model.utils import (
save_spectrogram,
)
from model.utils_infer import (
load_vocoder,
load_model,
preprocess_ref_audio_text,
infer_process,
remove_silence_for_generated_wav,
)
vocos = load_vocoder()