From 2973cd6835366b0abdad9b45360f7db7227f1fe8 Mon Sep 17 00:00:00 2001 From: Hacker 17082006 Date: Fri, 17 Feb 2023 22:08:29 +0700 Subject: [PATCH] Fix node bugs --- nodes.py | 4 ++-- requirements.txt | 8 +++++++- 2 files changed, 9 insertions(+), 3 deletions(-) diff --git a/nodes.py b/nodes.py index b0997b77a..aafbd31a3 100644 --- a/nodes.py +++ b/nodes.py @@ -222,7 +222,7 @@ class CannyPreprocessor: return {"required": { "image": ("IMAGE", ) , "low_threshold": ("INT", {"default": 100, "min": 0, "max": 255, "step": 1}), "high_threshold": ("INT", {"default": 100, "min": 0, "max": 255, "step": 1}), - "l2gradient": (["disable", "enable"]) + "l2gradient": (["disable", "enable"], ) }} RETURN_TYPES = ("IMAGE",) FUNCTION = "detect_edge" @@ -290,7 +290,7 @@ class OpenPosePreprocessor: @classmethod def INPUT_TYPES(s): return {"required": { "image": ("IMAGE", ), - "detect_hand": (["disable", "enable"]) + "detect_hand": (["disable", "enable"],) }} RETURN_TYPES = ("IMAGE",) FUNCTION = "estimate_pose" diff --git a/requirements.txt b/requirements.txt index e4be9ebc2..83baf6984 100644 --- a/requirements.txt +++ b/requirements.txt @@ -9,4 +9,10 @@ safetensors pytorch_lightning aiohttp accelerate - +opencv-contrib-python==4.3.0.36 +numpy=1.23.1 +timm==0.6.12 +torchvision=0.13.1 +scipy +matplotlib +skimage \ No newline at end of file