diff --git a/nodes/curve_nodes.py b/nodes/curve_nodes.py index 77be51f..053d8c6 100644 --- a/nodes/curve_nodes.py +++ b/nodes/curve_nodes.py @@ -1,14 +1,12 @@ import torch from torchvision import transforms import json -from PIL import Image, ImageDraw, ImageFont, ImageColor, ImageFilter, ImageChops +from PIL import Image, ImageDraw, ImageFont, ImageFilter import numpy as np from ..utility.utility import pil2tensor, tensor2pil import folder_paths import io import base64 - -from comfy.utils import common_upscale def parse_color(color): if isinstance(color, str) and ',' in color: diff --git a/nodes/mask_nodes.py b/nodes/mask_nodes.py index d9d6ad8..cd9db2d 100644 --- a/nodes/mask_nodes.py +++ b/nodes/mask_nodes.py @@ -881,6 +881,7 @@ class CreateVoronoiMask: def createvoronoi(self, frames, num_points, line_width, speed, frame_width, frame_height): from scipy.spatial import Voronoi + from matplotlib import pyplot as plt # Define the number of images in the batch batch_size = frames out = [] diff --git a/nodes/nodes.py b/nodes/nodes.py index da6a527..7a88a7f 100644 --- a/nodes/nodes.py +++ b/nodes/nodes.py @@ -3,7 +3,6 @@ import torch.nn as nn import numpy as np from PIL import Image import json, re, os, io, time -import re import importlib from comfy import model_management