Fix CheckpointLoader even though it is deprecated

This commit is contained in:
doctorpangloss 2024-01-29 17:20:10 -08:00
parent 0d2cc553bc
commit 32d83e52ff
14 changed files with 3 additions and 2 deletions

View File

@ -1,6 +1,7 @@
import os
import sys
import time
from pkg_resources import resource_filename
from ..cli_args import args
@ -21,8 +22,7 @@ else:
base_path = os.getcwd()
models_dir = os.path.join(base_path, "models")
folder_names_and_paths["checkpoints"] = ([os.path.join(models_dir, "checkpoints")], supported_pt_extensions)
folder_names_and_paths["configs"] = ([os.path.join(models_dir, "configs")], [".yaml"])
folder_names_and_paths["configs"] = ([os.path.join(models_dir, "configs"), resource_filename("comfy", "configs/")], [".yaml"])
folder_names_and_paths["loras"] = ([os.path.join(models_dir, "loras")], supported_pt_extensions)
folder_names_and_paths["vae"] = ([os.path.join(models_dir, "vae")], supported_pt_extensions)
folder_names_and_paths["clip"] = ([os.path.join(models_dir, "clip")], supported_pt_extensions)

View File

@ -8,6 +8,7 @@ import random
from PIL import Image, ImageOps, ImageSequence
from PIL.PngImagePlugin import PngInfo
from pkg_resources import resource_filename
import numpy as np
import safetensors.torch

View File