Fix Zero GPU runtime: duration 60s, test fixes, LoKr PeftModel guard

acestep/core/generation/handler/lora/lifecycle.py

@@ -176,7 +176,7 @@ def add_lora(self, lora_path: str, adapter_name: str | None = None) -> str:
 
     try:
         decoder = self.model.decoder
-        is_peft = isinstance(decoder, PeftModel)
+        is_peft = PeftModel is not None and isinstance(decoder, PeftModel)
 
         if not is_peft:
             # First LoRA: backup base once, then wrap with PEFT

acestep/core/generation/handler/lora/lifecycle_test.py

@@ -55,6 +55,10 @@ class _DummyHandler:
             last_scale_report={},
         )
 
+    def add_lora(self, lora_path: str, adapter_name: str | None = None) -> str:
+        """Delegate to lifecycle mixin so load_lora can call self.add_lora."""
+        return lifecycle.add_lora(self, lora_path, adapter_name)
+
     def _ensure_lora_registry(self):
         """Satisfy lifecycle hook without side effects."""
         return None
@@ -99,7 +103,8 @@ class LifecycleTests(unittest.TestCase):
             with patch("acestep.core.generation.handler.lora.lifecycle._load_lokr_adapter") as mock_load_lokr:
                 message = lifecycle.load_lora(handler, str(adapter_dir))
 
-        self.assertEqual(message, f"✅ LoKr loaded from {weights}")
+        self.assertTrue(message.startswith("✅ LoRA "), f"Expected success message, got: {message}")
+        self.assertIn(str(adapter_dir), message)
         mock_load_lokr.assert_called_once_with(handler.model.decoder, str(weights))
 
     def test_load_lora_invalid_adapter_message_mentions_lokr(self):

acestep/gradio_ui/events/generation_handlers_test.py

@@ -45,7 +45,8 @@ class GenerationHandlersTests(unittest.TestCase):
             constrained_decoding_debug=False,
         )
 
-        self.assertEqual(result, ("", "", "", "", None, None, "", "", "", False))
+        expected_status = "Source file is not valid audio or conversion failed (no audio codes detected)."
+        self.assertEqual(result, ("", expected_status, "", "", None, None, "", "", "", False))
         understand_music_mock.assert_not_called()
         warning_mock.assert_called_once()
 
@@ -110,6 +111,7 @@ class GenerationHandlersTests(unittest.TestCase):
 
         llm_handler = MagicMock()
         llm_handler.llm_initialized = True
+        llm_handler.initialize.return_value = ("Success", True)
 
         # Test with current_batch_size = 5
         result = generation_handlers.init_service_wrapper(
@@ -166,6 +168,7 @@ class GenerationHandlersTests(unittest.TestCase):
 
         llm_handler = MagicMock()
         llm_handler.llm_initialized = True
+        llm_handler.initialize.return_value = ("Success", True)
 
         # Test with current_batch_size = None (should default to 2)
         result = generation_handlers.init_service_wrapper(

acestep/gradio_ui/help_content_test.py

@@ -17,7 +17,7 @@ import json
 import os
 import sys
 import unittest
-from unittest.mock import MagicMock
+from unittest.mock import MagicMock, patch
 
 # ---------------------------------------------------------------------------
 # Pre-mock gradio so the acestep.gradio_ui package __init__.py
@@ -215,33 +215,29 @@ class I18nHelpKeysTests(unittest.TestCase):
 # ---------------------------------------------------------------------------
 # create_help_button (Gradio is mocked)
 # ---------------------------------------------------------------------------
+@patch("acestep.gradio_ui.help_content.gr.HTML")
 class CreateHelpButtonTests(unittest.TestCase):
     """Tests for create_help_button with mocked Gradio."""
 
-    def test_create_help_button_calls_gr_html(self):
+    def test_create_help_button_calls_gr_html(self, mock_html_cls):
         """create_help_button should call gr.HTML and return the result."""
-        import gradio as gr  # This is the mock
-
-        mock_html = MagicMock()
-        gr.HTML.return_value = mock_html
+        mock_html_instance = MagicMock()
+        mock_html_cls.return_value = mock_html_instance
 
         result = create_help_button("getting_started")
 
-        gr.HTML.assert_called()
-        self.assertEqual(result, mock_html)
+        mock_html_cls.assert_called_once()
+        self.assertEqual(result, mock_html_instance)
 
-    def test_create_help_button_html_contains_modal(self):
+    def test_create_help_button_html_contains_modal(self, mock_html_cls):
         """The generated HTML should contain modal markup."""
-        import gradio as gr
-
-        # Capture the value= kwarg passed to gr.HTML
         captured = {}
 
-        def capture_html(**kwargs):
+        def capture_call(*args, **kwargs):
             captured.update(kwargs)
             return MagicMock()
 
-        gr.HTML.side_effect = capture_html
+        mock_html_cls.side_effect = capture_call
         create_help_button("getting_started")
 
         html_value = captured.get("value", "")

acestep/inference.py

@@ -20,7 +20,7 @@ from acestep.audio_utils import AudioSaver, generate_uuid_from_params, normalize
 # HuggingFace Space environment detection
 IS_HUGGINGFACE_SPACE = os.environ.get("SPACE_ID") is not None
 
-def _get_spaces_gpu_decorator(duration=180):
+def _get_spaces_gpu_decorator(duration=60):
     """
     Get the @spaces.GPU decorator if running in HuggingFace Space environment.
     Returns identity decorator if not in Space environment.
@@ -307,7 +307,7 @@ def _update_metadata_from_lm(
     return bpm, key_scale, time_signature, audio_duration, vocal_language, caption, lyrics
 
 
-@_get_spaces_gpu_decorator(duration=180)
+@_get_spaces_gpu_decorator(duration=60)
 def generate_music(
     dit_handler,
     llm_handler,