1

app.py

@@ -25,10 +25,10 @@ HF_ENABLE = False
 LOCAL_DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
 
-def gpu_decorator(fn):
-    if IS_HF_SPACE and HF_ENABLE and spaces is not None:
-        return spaces.GPU(fn)
-    return fn
+# def gpu_decorator(fn):
+#     if IS_HF_SPACE and HF_ENABLE and spaces is not None:
+#         return spaces.GPU(fn)
+#     return fn
 
 
 def local_move2gpu(x):
@@ -45,7 +45,7 @@ _model = None
 _separator = None
 
 
-@gpu_decorator
+@spaces.GPU
 def get_model():
     """加载 YingMusicSinger 模型 / Load YingMusicSinger model."""
     download_files(task="infer")
@@ -59,7 +59,7 @@ def get_model():
     return _model
 
 
-@gpu_decorator
+@spaces.GPU
 def get_separator():
     """
     加载 MelBandRoformer 分离模型 / Load MelBandRoformer separator.
@@ -83,7 +83,7 @@ def get_separator():
 # ---------------------------------------------------------------------------
 # Vocal separation utilities / 人声分离工具
 # ---------------------------------------------------------------------------
-@gpu_decorator
+@spaces.GPU
 def separate_vocals(
     audio_path: str,
     device: str = "cuda:0",
@@ -151,7 +151,7 @@ def mix_vocal_and_accompaniment(
 # ---------------------------------------------------------------------------
 # Inference wrapper / 推理入口
 # ---------------------------------------------------------------------------
-@gpu_decorator
+@spaces.GPU
 def synthesize(
     ref_audio,
     melody_audio,