Update forge_photomaker.py

This commit is contained in:
lllyasviel
2024-02-02 14:08:04 -08:00
parent 98b0d7a999
commit d9c7616555

View File

@@ -1,11 +1,8 @@
from modules_forge.supported_preprocessor import PreprocessorClipVision, Preprocessor, PreprocessorParameter
from modules_forge.supported_preprocessor import Preprocessor, PreprocessorParameter
from modules_forge.shared import add_supported_preprocessor
from modules_forge.forge_util import numpy_to_pytorch
from modules_forge.shared import add_supported_control_model
from modules_forge.supported_controlnet import ControlModelPatcher
from ldm_patched.contrib.external_photomaker import PhotoMakerEncode, PhotoMakerIDEncoder
from ldm_patched.modules.sample import convert_cond
from ldm_patched.modules.samplers import encode_model_conds
opPhotoMakerEncode = PhotoMakerEncode().apply_photomaker