model-compose 0.4.36__tar.gz → 0.4.37__tar.gz

{model_compose-0.4.36 → model_compose-0.4.37}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: model-compose
-Version: 0.4.36
+Version: 0.4.37
 Summary: model-compose: Declarative AI Workflow Orchestrator
 Author-email: Hanyeol Cho <hanyeol.cho@gmail.com>
 License-Expression: MIT

{model_compose-0.4.36 → model_compose-0.4.37}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "model-compose"
-version = "0.4.36"
+version = "0.4.37"
 description = "model-compose: Declarative AI Workflow Orchestrator"
 authors = [
     { name = "Hanyeol Cho", email = "hanyeol.cho@gmail.com" }

{model_compose-0.4.36 → model_compose-0.4.37}/src/mindor/core/component/services/model/tasks/face_embedding/insightface.py

@@ -43,13 +43,11 @@ class InsightfaceFaceEmbeddingTaskService(FaceEmbeddingTaskService):
     def get_setup_requirements(self) -> Optional[List[str]]:
         return [ "insightface", "opencv-python", "onnxruntime" ]
 
-    async def _serve(self) -> None:
-        try:
-            self.model = self._load_pretrained_model()
-            logging.info(f"Model loaded successfully: {self.config.model}")
-        except Exception as e:
-            logging.error(f"Failed to load model '{self.config.model}': {e}")
-            raise
+    def _load_model(self) -> None:
+        self.model = self._load_pretrained_model()
+
+    def _unload_model(self) -> None:
+        self.model = None
 
     def _load_pretrained_model(self) -> FaceAnalysis:
         from insightface.app import FaceAnalysis

{model_compose-0.4.36 → model_compose-0.4.37}/src/mindor/core/component/services/model/tasks/image_generation/flux.py

@@ -40,15 +40,10 @@ class FluxImageGenerationTaskService(ImageGenerationTaskService):
     def get_setup_requirements(self) -> Optional[List[str]]:
         return None
 
-    async def _serve(self) -> None:
-        try:
-            self.pipeline, self.device = self._load_pretrained_pipeline()
-            logging.info(f"Model loaded successfully on device '{self.device}': {self.config.model}")
-        except Exception as e:
-            logging.error(f"Failed to load model '{self.config.model}': {e}")
-            raise
+    def _load_model(self) -> None:
+        self.pipeline, self.device = self._load_pretrained_pipeline()
 
-    async def _shutdown(self) -> None:
+    def _unload_model(self) -> None:
         self.pipeline = None
         self.device = None
 

{model_compose-0.4.36 → model_compose-0.4.37}/src/mindor/core/component/services/model/tasks/image_generation/hunyuan_image.py

@@ -40,15 +40,10 @@ class HunyuanImageGenerationTaskService(ImageGenerationTaskService):
     def get_setup_requirements(self) -> Optional[List[str]]:
         return None
 
-    async def _serve(self) -> None:
-        try:
-            self.pipeline, self.device = self._load_pretrained_pipeline()
-            logging.info(f"Model loaded successfully on device '{self.device}': {self.config.model}")
-        except Exception as e:
-            logging.error(f"Failed to load model '{self.config.model}': {e}")
-            raise
+    def _load_model(self) -> None:
+        self.pipeline, self.device = self._load_pretrained_pipeline()
 
-    async def _shutdown(self) -> None:
+    def _unload_model(self) -> None:
         self.pipeline = None
         self.device = None
 

{model_compose-0.4.36 → model_compose-0.4.37}/src/mindor/core/component/services/model/tasks/image_generation/sdxl.py

@@ -40,15 +40,10 @@ class SdxlImageGenerationTaskService(ImageGenerationTaskService):
     def get_setup_requirements(self) -> Optional[List[str]]:
         return None
 
-    async def _serve(self) -> None:
-        try:
-            self.pipeline, self.device = self._load_pretrained_pipeline()
-            logging.info(f"Model loaded successfully on device '{self.device}': {self.config.model}")
-        except Exception as e:
-            logging.error(f"Failed to load model '{self.config.model}': {e}")
-            raise
+    def _load_model(self) -> None:
+        self.pipeline, self.device = self._load_pretrained_pipeline()
 
-    async def _shutdown(self) -> None:
+    def _unload_model(self) -> None:
         self.pipeline = None
         self.device = None
 

{model_compose-0.4.36 → model_compose-0.4.37}/src/mindor/core/component/services/model/tasks/image_to_text/huggingface.py

@@ -59,7 +59,7 @@ class HuggingfaceImageToTextTaskAction:
             batch_images = images[index:index + batch_size]
             batch_texts = texts[index:index + batch_size] if texts else None
 
-            inputs: Tensor = self.processor(images=batch_images, texts=batch_texts, **processor_params)
+            inputs: Tensor = self.processor(images=batch_images, text=batch_texts, **processor_params)
             inputs = inputs.to(self.device)
 
             def _generate():
@@ -71,7 +71,7 @@ class HuggingfaceImageToTextTaskAction:
                         streamer=streamer
                     )
 
-                outputs = self.processor.tokenizer.batch_decode(outputs)
+                outputs = self.processor.tokenizer.batch_decode(outputs, skip_special_tokens=True)
                 results.extend(outputs)
 
             if streaming:

{model_compose-0.4.36 → model_compose-0.4.37}/src/mindor/core/component/services/model/tasks/image_upscale/esrgan.py

@@ -41,15 +41,10 @@ class EsrganImageUpscaleTaskService(ImageUpscaleTaskService):
     def get_setup_requirements(self) -> Optional[List[str]]:
         return [ "basicsr", "torchvision" ]
 
-    async def _serve(self) -> None:
-        try:
-            self.model, self.device = self._load_pretrained_model()
-            logging.info(f"Model loaded successfully on device '{self.device}': {self.config.model}")
-        except Exception as e:
-            logging.error(f"Failed to load model '{self.config.model}': {e}")
-            raise
+    def _load_model(self) -> None:
+        self.model, self.device = self._load_pretrained_model()
 
-    async def _shutdown(self) -> None:
+    def _unload_model(self) -> None:
         self.model = None
         self.device = None
 

{model_compose-0.4.36 → model_compose-0.4.37}/src/mindor/core/component/services/model/tasks/image_upscale/ldsr.py

@@ -89,16 +89,11 @@ class LdsrImageUpscaleTaskService(ImageUpscaleTaskService):
     def get_setup_requirements(self) -> Optional[List[str]]:
         return None
 
-    async def _serve(self) -> None:
-        try:
-            self.model, self.device = self._load_pretrained_model()
-            logging.info(f"Model loaded successfully on device '{self.device}': {self.config.model}")
-        except Exception as e:
-            logging.error(f"Failed to load model '{self.config.model}': {e}")
-            raise
-
-    async def _shutdown(self) -> None:
-        self.model  = None
+    def _load_model(self) -> None:
+        self.model, self.device = self._load_pretrained_model()
+
+    def _unload_model(self) -> None:
+        self.model = None
         self.device = None
 
     def _load_pretrained_model(self) -> Tuple[PreTrainedModel, torch.device]:

{model_compose-0.4.36 → model_compose-0.4.37}/src/mindor/core/component/services/model/tasks/image_upscale/real_esrgan.py

@@ -64,15 +64,10 @@ class RealEsrganImageUpscaleTaskService(ImageUpscaleTaskService):
     def get_setup_requirements(self) -> Optional[List[str]]:
         return [ "realesrgan>=1.0@git+https://github.com/sberbank-ai/Real-ESRGAN.git" ]
 
-    async def _serve(self) -> None:
-        try:
-            self.model, self.device = self._load_pretrained_model()
-            logging.info(f"Model loaded successfully on device '{self.device}': {self.config.model}")
-        except Exception as e:
-            logging.error(f"Failed to load model '{self.config.model}': {e}")
-            raise
-
-    async def _shutdown(self) -> None:
+    def _load_model(self) -> None:
+        self.model, self.device = self._load_pretrained_model()
+
+    def _unload_model(self) -> None:
         self.model = None
         self.device = None
 

{model_compose-0.4.36 → model_compose-0.4.37}/src/mindor/core/component/services/model/tasks/image_upscale/swinir.py

@@ -165,15 +165,10 @@ class SwinIRImageUpscaleTaskService(ImageUpscaleTaskService):
     def get_setup_requirements(self) -> Optional[List[str]]:
         return [ "basicsr", "torch", "torchvision" ]
 
-    async def _serve(self) -> None:
-        try:
-            self.model, self.device = self._load_pretrained_model()
-            logging.info(f"Model loaded successfully on device '{self.device}': {self.config.model}")
-        except Exception as e:
-            logging.error(f"Failed to load model '{self.config.model}': {e}")
-            raise
-
-    async def _shutdown(self) -> None:
+    def _load_model(self) -> None:
+        self.model, self.device = self._load_pretrained_model()
+
+    def _unload_model(self) -> None:
         self.model = None
         self.device = None
 

{model_compose-0.4.36 → model_compose-0.4.37}/src/mindor/core/component/services/model/tasks/text_generation/huggingface.py

@@ -65,7 +65,7 @@ class HuggingfaceTextGenerationTaskAction:
                     )
 
                 if not streaming:
-                    outputs = self.tokenizer.batch_decode(outputs)
+                    outputs = self.tokenizer.batch_decode(outputs, skip_special_tokens=True)
                     results.extend(outputs)
 
             if streaming:

{model_compose-0.4.36 → model_compose-0.4.37}/src/mindor/core/component/services/model/tasks/text_to_speech/qwen.py

@@ -95,15 +95,10 @@ class QwenTextToSpeechTaskService(TextToSpeechTaskService):
     def get_setup_requirements(self) -> Optional[List[str]]:
         return [ "transformers", "qwen_tts", "soundfile" ]
 
-    async def _serve(self) -> None:
-        try:
-            self.model, self.device = self._load_pretrained_model()
-            logging.info(f"Model loaded successfully on device '{self.device}': {self.config.model}")
-        except Exception as e:
-            logging.error(f"Failed to load model '{self.config.model}': {e}")
-            raise
-
-    async def _shutdown(self) -> None:
+    def _load_model(self) -> None:
+        self.model, self.device = self._load_pretrained_model()
+
+    def _unload_model(self) -> None:
         self.model = None
         self.device = None
 

{model_compose-0.4.36 → model_compose-0.4.37}/src/mindor/core/controller/webui/gradio/builder.py

@@ -98,6 +98,11 @@ class GradioWebUIBuilder:
                     yield [ _run_button_running(), *self._build_interrupt_updates(output), *(gr.update() for _ in output_components) ]
                     return
 
+                # Check if the result is a failed TaskResult
+                if isinstance(output, dict) and output.get("status") == "failed":
+                    yield [ _run_button_ready(), *self._clear_interrupt_updates(), *(gr.update() for _ in output_components) ]
+                    raise gr.Error(str(output.get("error", "Workflow failed")))
+
                 # Clear interrupt panel for normal results
                 clear_interrupt = self._clear_interrupt_updates()
 
@@ -142,7 +147,9 @@ class GradioWebUIBuilder:
 
                 clear_interrupt = self._clear_interrupt_updates()
 
+                # Check if the result is a failed TaskResult
                 if result.get("status") == "failed":
+                    yield [ _run_button_ready(), _resume_button_ready(), *self._clear_interrupt_updates(), *(gr.update() for _ in output_components) ]
                     raise gr.Error(str(result.get("error", "Workflow failed")))
 
                 # Completed — fetch output
@@ -250,6 +257,14 @@ class GradioWebUIBuilder:
 
         return value if value != "" else None
 
+    async def _format_workflow_output(self, output: Any, workflow: WorkflowSchema) -> Optional[list]:
+        if workflow.output and output is not None:
+            output = await self._flatten_output_value(output, workflow.output)
+        if output is None:
+            return None
+        result = output[0] if isinstance(output, (list, tuple)) and len(output) == 1 else output
+        return list(result) if isinstance(result, (list, tuple)) else [result]
+
     def _build_interrupt_components(self):
         message  = gr.Markdown("")
         metadata = gr.JSON(label="Metadata", visible=False)

{model_compose-0.4.36 → model_compose-0.4.37}/src/model_compose.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: model-compose
-Version: 0.4.36
+Version: 0.4.37
 Summary: model-compose: Declarative AI Workflow Orchestrator
 Author-email: Hanyeol Cho <hanyeol.cho@gmail.com>
 License-Expression: MIT