Reorganize and Cleanup

* Moved files into more standard ComfyUI extension layout
* Removed need for syspath change
* Removed unused imports
This commit is contained in:
Michael Poutre
2023-10-24 14:55:04 -07:00
parent 62175f95b2
commit 06e8df4ef6
3 changed files with 8 additions and 44 deletions
-3
View File
@@ -1,3 +0,0 @@
from .LCM_Nodes import NODE_CLASS_MAPPINGS
__all__ = ['NODE_CLASS_MAPPINGS']
+5 -41
View File
@@ -1,56 +1,20 @@
from tkinter import Y
import torch
from PIL import Image, ImageDraw, ImageFont, ImageEnhance, ImageOps, ImageFilter
import PIL
import torch.nn.functional as F
import cv2
import os
import torch
import sys
from PIL import ImageDraw, ImageOps, ImageFilter
import json
import hashlib
import traceback
import time
import re
import glob
from PIL.PngImagePlugin import PngInfo
import numpy as np
import safetensors.torch
import random
sys.path.insert(0, os.path.join(os.path.dirname(os.path.realpath(__file__)), "comfy"))
import comfy.diffusers_load
import comfy.samplers
import comfy.sample
import comfy.sd
import comfy.utils
import comfy.clip_vision
import comfy.model_management
import importlib
import folder_paths
import latent_preview
import math
import struct
from .lcm_pipeline_inpaint import LatentConsistencyModelPipeline_inpaint, LCMScheduler_X
from .LCM.lcm_pipeline_inpaint import LatentConsistencyModelPipeline_inpaint, LCMScheduler_X
from diffusers import AutoencoderKL, UNet2DConditionModel
#from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker
from transformers import CLIPTokenizer, CLIPTextModel, CLIPImageProcessor
import os
import torch
from tqdm import tqdm
from safetensors.torch import load_file
from compel import Compel
import argparse
import random
from PIL import Image
import tomesd
from comfy.cli_args import args
class LCMLoader:
def __init__(self):
+3
View File
@@ -0,0 +1,3 @@
from .LCM_Nodes import NODE_CLASS_MAPPINGS
__all__ = ['NODE_CLASS_MAPPINGS']