bizydraft 0.2.29__py3-none-any.whl → 0.2.31__py3-none-any.whl

bizydraft/hijack_nodes.py

@@ -53,6 +53,18 @@ DATA_LOAD_CLASSES = [
     "LoadImageMask",
     "LoadAudio",
     "Load3D",
+    "LayerMask: YoloV8Detect",
+    "Lora Loader Stack (rgthree)",
+    "easy loraNames",
+    "easy loraStack",
+    "Load Lora",
+    "Intrinsic_lora_sampling",
+    "ADE_LoadAnimateDiffModel",
+    "ADE_AnimateDiffLoRALoader",
+    "easy ultralyticsDetectorPipe",
+    "UltralyticsDetectorProvider",
+    "SAMLoader",
+    "easy samLoaderPipe",
 ]
 
 

bizydraft/prestartup_patch.py

@@ -19,3 +19,16 @@ def prestartup_patch():
         torch.cuda.current_device = lambda: 0
         torch.cuda.device_count = lambda: 1
         torch.cuda.get_device_name = lambda x=0: "Fake CUDA Device"
+        torch.cuda.get_device_properties = lambda x=0: type(
+            "FakeProps",
+            (),
+            {
+                "name": "Fake CUDA Device",
+                "major": 8,
+                "minor": 0,
+                "total_memory": 8589934592,
+                "multi_processor_count": 80,
+                "uuid": "019f4533-ac25-1db6-21b9-97977befe1ef",
+                "L2_cache_size": "60MB",
+            },
+        )()

bizydraft/static/js/hookLoadModel.js

@@ -14,7 +14,12 @@ const WHITELIST_NODES = {
     "ADE_AnimateDiffLoRALoader": "LoRa",
     "easy ultralyticsDetectorPipe": "Detection",
     "UltralyticsDetectorProvider": "Detection",
+    "SAMLoader": "Detection",
+    "easy samLoaderPipe": "Detection",
 };
+const HAZY_WHITELIST_NODES = {
+
+}
 
 const possibleWidgetNames=[
     "clip_name",
@@ -62,6 +67,13 @@ function getModelTypeFromNodeName(nodeName) {
         console.log(`白名单匹配: ${nodeName} -> ${WHITELIST_NODES[nodeName]}`);
         return WHITELIST_NODES[nodeName];
     }
+    const HAZY_WHITELIST_NODES_KEYS = Object.keys(HAZY_WHITELIST_NODES);
+    const thisKey = HAZY_WHITELIST_NODES_KEYS.find(key => nodeName.toLowerCase().includes(key.toLowerCase()));
+    if (thisKey) {
+        console.log(`模糊白名单匹配: ${nodeName} -> ${HAZY_WHITELIST_NODES[thisKey]}`);
+        return HAZY_WHITELIST_NODES[thisKey];
+    }
+
 
     // 然后使用正则表达式匹配
     const regex = /^(\w+).*Loader.*/i;

{bizydraft-0.2.29.dist-info → bizydraft-0.2.31.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bizydraft
-Version: 0.2.29
+Version: 0.2.31
 Summary: bizydraft
 Requires-Dist: loguru
 Requires-Dist: aiohttp

{bizydraft-0.2.29.dist-info → bizydraft-0.2.31.dist-info}/RECORD

@@ -1,12 +1,12 @@
 bizydraft/__init__.py,sha256=OM-sKCQrPh25nHVJIX-DgF1raMYyoWLSuyduIAHt0Gs,78
 bizydraft/block_nodes.py,sha256=Lqn3oSCaGDHR2OICc8a2iRoRCVVK9v1-9MM3r-qIZgA,1092
 bizydraft/env.py,sha256=nVh4xBKXWnRUPbHL3PqZr6VFN8txKsrsmshu8PFGaqo,299
-bizydraft/hijack_nodes.py,sha256=4msT6YJqC_bQXKPmm76wiSJIhkZXETsGm1jz4qHHcZ4,2567
+bizydraft/hijack_nodes.py,sha256=6k2MvC0lhzg32zr14twuUMnAeYxPvUYWL2ID6Tv8cw8,2904
 bizydraft/hijack_routes.py,sha256=lf6x3xDzbo9yQIRwfG_1oxcUNfrX_1ogbiff3WOV9gM,3268
 bizydraft/oss_utils.py,sha256=W3-Td6K4vrB83oMowN82c3nmQ3IrEoR1UX6S8AvO6ig,7896
 bizydraft/patch_handlers.py,sha256=WPf4xeV5sBpoJ2MkHHhx0wem7B0KQNuhONElN7c9v3g,5989
 bizydraft/postload.py,sha256=XFElKcmCajT_oO7SVJJBaN04XcWro54N5HB5cSCxfvI,1308
-bizydraft/prestartup_patch.py,sha256=sNXzispEYlR5u3yJvL7H1Z9otyx50_iRFNNWU0p7K4g,561
+bizydraft/prestartup_patch.py,sha256=4FGjmRcDHELjtlQOrfTfk2Un5OS89QIqfq-gEcB9WDs,998
 bizydraft/resp.py,sha256=8INvKOe5Dgai3peKfqKjrhUoYeuXWXn358w30-_cY-A,369
 bizydraft/server.py,sha256=L2zoJgOisr65IRphOyko74AdsLel59gh55peyMaUrO8,2102
 bizydraft/workflow_io.py,sha256=MYhJbpgkv8hrA5k_aolijOTrWpTtu62nzRznA4hv8JE,4298
@@ -14,7 +14,7 @@ bizydraft/static/js/aiAppHandler.js,sha256=t3tefGg9d_2-VeNxqrwMMdX6JYE5uroyXZq0h
 bizydraft/static/js/freezeModeHandler.js,sha256=l1-JK8hyFbJzQ6ylq6occUa_kwszPq9rDbKut48bn6s,17247
 bizydraft/static/js/handleStyle.js,sha256=o9JpjOqs_lLIr-jpqpGNjUN74XQPT8pD3fF5bjsHLAU,1466
 bizydraft/static/js/hookLoadImage.js,sha256=jQGckFr7bTLRiwcFsVmSy1mj_o7TKj1uup0QQ00qwbk,8079
-bizydraft/static/js/hookLoadModel.js,sha256=IitADQQ5eyo_69tFZ_D9XnJZ6T-gglac9cjdhc-Rwsw,8128
+bizydraft/static/js/hookLoadModel.js,sha256=j_U-z275xxeu3XqgSVlrP8jKURWni5yUaG3pcJQ3ZGI,8582
 bizydraft/static/js/main.js,sha256=oEsVEUZSo8ipx93oqs2WFQSRgp46XQ_D-Xao-wen2S8,121
 bizydraft/static/js/nodeFocusHandler.js,sha256=24xXbS4Q-GjJdRqf11i-1pBo8MkOJ24F7MHFV44EG6Q,4683
 bizydraft/static/js/nodeParamsFilter.js,sha256=H7lBB0G8HNqoGhOCH1hNXqPU-rPlrFyTxg_f_JgLEMk,4168
@@ -22,7 +22,7 @@ bizydraft/static/js/postEvent.js,sha256=5rafG9GjaesxmFD_8eSJQYHeSNWfOG3iWXLUPuaR
 bizydraft/static/js/socket.js,sha256=VE3fTAgEfM0FZhL526Skt7OCRokOa3mzTCAjAomI_tE,2432
 bizydraft/static/js/tool.js,sha256=dwVFLrT0pEsdOBkh2RRUDvyVo8GzOIVYg7vU5RiM0m0,2765
 bizydraft/static/js/uploadFile.js,sha256=WvglKzHMeOzDhOH3P-fLcPHxCLbKOJpo4DntoRxeJtI,4908
-bizydraft-0.2.29.dist-info/METADATA,sha256=2tNiwxHilsJEOvEkDmkA0Q9JhN0wdXJ-aX5uUrVVWqE,162
-bizydraft-0.2.29.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-bizydraft-0.2.29.dist-info/top_level.txt,sha256=XtoBq6hjZhXIM7aas4GtPDtAiKo8FdLzMABXW8qqQ8M,10
-bizydraft-0.2.29.dist-info/RECORD,,
+bizydraft-0.2.31.dist-info/METADATA,sha256=8Hfs60uIUxadzzAuwZnIHzCwMtWqjV33z40yDVTpCa8,162
+bizydraft-0.2.31.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+bizydraft-0.2.31.dist-info/top_level.txt,sha256=XtoBq6hjZhXIM7aas4GtPDtAiKo8FdLzMABXW8qqQ8M,10
+bizydraft-0.2.31.dist-info/RECORD,,