Remove unnecessary imports

This commit is contained in:
Limitex
2024-01-02 21:58:23 +09:00
parent ce97b72ead
commit 479139541c
2 changed files with 8 additions and 17 deletions
+5 -9
View File
@@ -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):
+3 -8
View File
@@ -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,