将自定义节点放置到custom目录下
This commit is contained in:
@@ -13,7 +13,6 @@ class SaveImagePath:
|
||||
"image_path": ("IMAGE", {"forceInput": True}),
|
||||
}
|
||||
}
|
||||
|
||||
RETURN_TYPES = ("STRING",)
|
||||
FUNCTION = "load"
|
||||
CATEGORY = "不忘科技-自定义节点🚩"
|
||||
@@ -50,13 +49,13 @@ class SaveImagePath:
|
||||
|
||||
pil_image = Image.fromarray(image_path)
|
||||
|
||||
# output_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "output")
|
||||
output_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "output")
|
||||
# base_dir = os.path.dirname(os.path.abspath(__file__))
|
||||
output_dir = '/root/comfy/ComfyUI/output'
|
||||
# output_dir = '/root/comfy/ComfyUI/output'
|
||||
if not os.path.exists(output_dir):
|
||||
os.makedirs(output_dir)
|
||||
|
||||
file_name = "%s.jpg" % str(uuid.uuid4())
|
||||
file_name = "%s.png" % str(uuid.uuid4())
|
||||
p = os.path.join(output_dir, file_name)
|
||||
pil_image.save(p)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user