xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2026-06-09 20:24:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

minor cleanup

Browse Source
This commit is contained in:
kijai 2025-12-09 07:55:43 +02:00
parent 022b499c70
commit 91a141c8b8
3 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
nodes/curve_nodes.py
View File

@ -1,15 +1,13 @@
import torch import torch
from torchvision import transforms from torchvision import transforms
import json import json
from PIL import Image, ImageDraw, ImageFont, ImageColor, ImageFilter, ImageChops from PIL import Image, ImageDraw, ImageFont, ImageFilter
import numpy as np import numpy as np
from ..utility.utility import pil2tensor, tensor2pil from ..utility.utility import pil2tensor, tensor2pil
import folder_paths import folder_paths
import io import io
import base64 import base64
from comfy.utils import common_upscale
def parse_color(color): def parse_color(color):
if isinstance(color, str) and ',' in color: if isinstance(color, str) and ',' in color:
return tuple(int(c.strip()) for c in color.split(',')) return tuple(int(c.strip()) for c in color.split(','))

1
nodes/mask_nodes.py
View File

@ -881,6 +881,7 @@ class CreateVoronoiMask:
def createvoronoi(self, frames, num_points, line_width, speed, frame_width, frame_height): def createvoronoi(self, frames, num_points, line_width, speed, frame_width, frame_height):
from scipy.spatial import Voronoi from scipy.spatial import Voronoi
from matplotlib import pyplot as plt
# Define the number of images in the batch # Define the number of images in the batch
batch_size = frames batch_size = frames
out = [] out = []

1
nodes/nodes.py
View File

@ -3,7 +3,6 @@ import torch.nn as nn
import numpy as np import numpy as np
from PIL import Image from PIL import Image
import json, re, os, io, time import json, re, os, io, time
import re
import importlib import importlib
from comfy import model_management from comfy import model_management