Refactor MochiVideoAPI initialization and update logging

- Simplified the initialization of MochiSettings in the MochiVideoAPI class by removing specific parameters, allowing for default settings. - Enhanced logging in combined_api.log to include multiple server start entries, improving traceability of server activity. These changes aim to streamline the API initialization process and improve logging clarity for better monitoring of the video generation server.
VikramxD · Dec 23, 2024 · 354cf81 · 354cf81
1 parent 07754af
commit 354cf81
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 6 deletions.
diff --git a/api/__pycache__/mochi_serve.cpython-310.pyc b/api/__pycache__/mochi_serve.cpython-310.pyc
diff --git a/api/logs/combined_api.log b/api/logs/combined_api.log
@@ -1,2 +1,4 @@
 2024-12-23 14:03:00.055 | INFO     | __main__:main:220 - Starting Combined Video Generation Server
 2024-12-23 14:58:34.285 | INFO     | __main__:main:278 - Starting combined video generation server on port 8000
+2024-12-23 15:06:16.741 | INFO     | __main__:main:278 - Starting combined video generation server on port 8000
+2024-12-23 15:07:54.567 | INFO     | __main__:main:278 - Starting combined video generation server on port 8000
diff --git a/api/mochi_serve.py b/api/mochi_serve.py
@@ -134,12 +134,7 @@ def setup(self, device: str) -> None:
         Raises:
             RuntimeError: On initialization failure
         """
-        self.settings = MochiSettings(
-            model_name="Mini-Mochi",
-            enable_vae_tiling=True,
-            enable_attention_slicing=True,
-            device=device
-        )
+        self.settings = MochiSettings()
 
         logger.info("Initializing Mochi inference engine")
         self.engine = MochiInference(self.settings)