bizyengine 1.2.10__tar.gz → 1.2.11__tar.gz

{bizyengine-1.2.10 → bizyengine-1.2.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bizyengine
-Version: 1.2.10
+Version: 1.2.11
 Summary: [a/BizyAir](https://github.com/siliconflow/BizyAir) Comfy Nodes that can run in any environment.
 Author-email: SiliconFlow <yaochi@siliconflow.cn>
 Project-URL: Repository, https://github.com/siliconflow/BizyAir

{bizyengine-1.2.10 → bizyengine-1.2.11}/bizyengine/misc/llm.py

@@ -318,6 +318,29 @@ class BizyAirJoyCaption2(BizyAirMiscBaseNode):
 
     CATEGORY = "☁️BizyAir/AI Assistants"
 
+    def resize_if_large(self, image, max_size=384):
+        import torch.nn.functional as F
+
+        _, h, w, _ = image.shape
+
+        if h <= max_size and w <= max_size:
+            return image
+
+        if h > w:
+            new_h = max_size
+            new_w = int(w * (max_size / h))
+        else:
+            new_w = max_size
+            new_h = int(h * (max_size / w))
+
+        image = image.permute(0, 3, 1, 2)
+        resized_image = F.interpolate(
+            image, size=(new_h, new_w), mode="bilinear", align_corners=False
+        )
+        resized_image = resized_image.permute(0, 2, 3, 1)
+
+        return resized_image
+
     def joycaption2(
         self,
         image,
@@ -333,12 +356,7 @@ class BizyAirJoyCaption2(BizyAirMiscBaseNode):
     ):
         extra_data = pop_api_key_and_prompt_id(kwargs)
         headers = client.headers(api_key=extra_data["api_key"])
-
-        SIZE_LIMIT = 1536
-        _, w, h, c = image.shape
-        assert (
-            w <= SIZE_LIMIT and h <= SIZE_LIMIT
-        ), f"width and height must be less than {SIZE_LIMIT}x{SIZE_LIMIT}, but got {w} and {h}"
+        image = self.resize_if_large(image)
 
         payload = {
             "image": None,
@@ -380,15 +398,21 @@ class BizyAirJoyCaption2(BizyAirMiscBaseNode):
         return (caption,)
 
 
+class BizyAirJoyCaption3(BizyAirJoyCaption2):
+    API_URL = f"{BIZYAIR_SERVER_ADDRESS}/supernode/joycaption3"
+
+
 NODE_CLASS_MAPPINGS = {
     "BizyAirSiliconCloudLLMAPI": SiliconCloudLLMAPI,
     "BizyAirSiliconCloudVLMAPI": SiliconCloudVLMAPI,
     "BizyAirJoyCaption": BizyAirJoyCaption,
     "BizyAirJoyCaption2": BizyAirJoyCaption2,
+    "BizyAirJoyCaption3": BizyAirJoyCaption3,
 }
 NODE_DISPLAY_NAME_MAPPINGS = {
     "BizyAirSiliconCloudLLMAPI": "☁️BizyAir SiliconCloud LLM API",
     "BizyAirSiliconCloudVLMAPI": "☁️BizyAir SiliconCloud VLM API",
     "BizyAirJoyCaption": "☁️BizyAir Joy Caption",
     "BizyAirJoyCaption2": "☁️BizyAir Joy Caption2",
+    "BizyAirJoyCaption3": "☁️BizyAir Joy Caption3",
 }

bizyengine-1.2.11/bizyengine/version.txt

@@ -0,0 +1 @@
+1.2.11

{bizyengine-1.2.10 → bizyengine-1.2.11}/bizyengine.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bizyengine
-Version: 1.2.10
+Version: 1.2.11
 Summary: [a/BizyAir](https://github.com/siliconflow/BizyAir) Comfy Nodes that can run in any environment.
 Author-email: SiliconFlow <yaochi@siliconflow.cn>
 Project-URL: Repository, https://github.com/siliconflow/BizyAir

bizyengine-1.2.10/bizyengine/version.txt

@@ -1 +0,0 @@
-1.2.10