From 479139541c9066fb74a512beff118e5c889ce3aa Mon Sep 17 00:00:00 2001
From: Limitex <76650151+Limitex@users.noreply.github.com>
Date: Tue, 2 Jan 2024 21:58:23 +0900
Subject: [PATCH] Remove unnecessary imports

---
 nodes.py | 14 +++++---------
 utils.py | 11 +++--------
 2 files changed, 8 insertions(+), 17 deletions(-)

diff --git a/nodes.py b/nodes.py
index 590b30f..84d3edf 100644
--- a/nodes.py
+++ b/nodes.py
@@ -1,18 +1,14 @@
-import json
 import copy
 import os
-from .utils import SCHEDULERS, convert_images_to_tensors, token_auto_concat_embeds, vae_pt_to_vae_diffuser
-import numpy as np
 import torch
-from comfy.model_management import get_torch_device, get_torch_device_name
+from safetensors.torch import load_file
+from .utils import SCHEDULERS, convert_images_to_tensors, token_auto_concat_embeds, vae_pt_to_vae_diffuser
+from comfy.model_management import get_torch_device
 import folder_paths
-from diffusers import StableDiffusionPipeline, AutoencoderKL, AutoencoderTiny
-from comfy.cli_args import args
-from PIL import Image, ImageOps, ImageSequence
-from PIL.PngImagePlugin import PngInfo
 from streamdiffusion import StreamDiffusion
 from streamdiffusion.image_utils import postprocess_image
-from safetensors.torch import load_file
+from diffusers import StableDiffusionPipeline, AutoencoderKL, AutoencoderTiny
+
 
 class DiffusersPipelineLoader:
     def __init__(self):
diff --git a/utils.py b/utils.py
index 3262453..59f5c9c 100644
--- a/utils.py
+++ b/utils.py
@@ -1,9 +1,9 @@
-import torch
-import argparse
 import io
-
+import torch
 import requests
+import numpy as np
 from omegaconf import OmegaConf
+from torchvision.transforms import ToTensor
 from diffusers.pipelines.stable_diffusion.convert_from_ckpt import (
     assign_to_checkpoint,
     conv_attn_to_linear,
@@ -11,11 +11,6 @@ from diffusers.pipelines.stable_diffusion.convert_from_ckpt import (
     renew_vae_attention_paths,
     renew_vae_resnet_paths,
 )
-from PIL import Image
-from torchvision.transforms import ToTensor
-import numpy as np
-
-
 from diffusers import (
     AutoencoderKL,
     DDIMScheduler,