bizyengine 1.2.10__py3-none-any.whl → 1.2.11__py3-none-any.whl

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/version.txt

@@ -1 +1 @@
-1.2.10
+1.2.11

{bizyengine-1.2.10.dist-info → bizyengine-1.2.11.dist-info}/METADATA

@@ -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.dist-info → bizyengine-1.2.11.dist-info}/RECORD

@@ -1,5 +1,5 @@
 bizyengine/__init__.py,sha256=GP9V-JM07fz7uv_qTB43QEA2rKdrVJxi5I7LRnn_3ZQ,914
-bizyengine/version.txt,sha256=QwQnXzDgdsdqvcDygPD91VN_GscAlTuhkI4ppk1TGRk,7
+bizyengine/version.txt,sha256=NzEMfzSBZRIsy59KhUwpSFOVuzrwZZulp3vgZsFRFnk,7
 bizyengine/bizy_server/__init__.py,sha256=SP9oSblnPo4KQyh7yOGD26YCskFAcQHAZy04nQBNRIw,200
 bizyengine/bizy_server/api_client.py,sha256=F2kSe5FVQz5b47-QzSH5kB9S-wP921MPWrtVbbQ3nEY,43623
 bizyengine/bizy_server/errno.py,sha256=Q-U96XnZQCuPH_44Om8wnc2-Kh7qFqwLKtox27msU54,16095
@@ -66,7 +66,7 @@ bizyengine/core/path_utils/path_manager.py,sha256=tRVAcpsYvfWD-tK7khLvNCZayB0wpU
 bizyengine/core/path_utils/utils.py,sha256=SY5TDoOV5Sr_n3Ez2xbcMRsll1A1J7hPZ8Zn4tpjF3c,2393
 bizyengine/misc/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bizyengine/misc/auth.py,sha256=V7VoHZ9-ljT_gUUOOh7AeNnAyVuyiK8fiqSM8T8k948,2671
-bizyengine/misc/llm.py,sha256=JOwzioAGtY1YjQZZfnm5xCB3frbbvNi3sHUltd_Qq2M,12856
+bizyengine/misc/llm.py,sha256=4fCMj5rxrjQ12Hyb5-JNHQHnugNdHoYZB2_9x8um8Lw,13526
 bizyengine/misc/mzkolors.py,sha256=mWQh2XDdRza1iNacGZ3YApqbs5RHrW0nDIPGN1ZfC8g,2826
 bizyengine/misc/nodes.py,sha256=HhLfYc_8Z7akcmtrvOviYdZ_GG5KhF5eoSjCzMK4pq4,43654
 bizyengine/misc/nodes_controlnet_aux.py,sha256=Tza7XgStfmTm29bVl7qupE--ELTDfk8JaGFb93DxL_Q,16398
@@ -75,7 +75,7 @@ bizyengine/misc/route_sam.py,sha256=-bMIR2QalfnszipGxSxvDAHGJa5gPSrjkYPb5baaRg4,
 bizyengine/misc/segment_anything.py,sha256=m-G0PsuP6IgpdUzhEqiqzPg1_svMVH5MTivOsoVR1-Y,8940
 bizyengine/misc/supernode.py,sha256=ox1_ufOFbm4zlpPQD8TGd0JKv94vsscu1cCJOdbaZPY,4531
 bizyengine/misc/utils.py,sha256=SkU_qHU3n3dnzq68mdg_E8380ivRDVZwIb4UgnakhMM,6828
-bizyengine-1.2.10.dist-info/METADATA,sha256=fkMBENG0KzaDDVH-pkmvcRIxxi23H9YSVMEsM6fPQBc,4210
-bizyengine-1.2.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-bizyengine-1.2.10.dist-info/top_level.txt,sha256=2zapzqxX-we5cRyJkGf9bd5JinRtXp3-_uDI-xCAnc0,11
-bizyengine-1.2.10.dist-info/RECORD,,
+bizyengine-1.2.11.dist-info/METADATA,sha256=JBQQgsJaoiTpf6L_zWoxjubKBm1APGJ3uSibg-rkkRI,4210
+bizyengine-1.2.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+bizyengine-1.2.11.dist-info/top_level.txt,sha256=2zapzqxX-we5cRyJkGf9bd5JinRtXp3-_uDI-xCAnc0,11
+bizyengine-1.2.11.dist-info/RECORD,,