minor cleanup

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

View File

@ -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:

View File

@ -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 = []

View File

@ -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