|
import os |
|
from comfy.sd import load_checkpoint_guess_config |
|
|
|
from nodes import ( |
|
VAEDecode, |
|
KSamplerAdvanced, |
|
EmptyLatentImage, |
|
SaveImage, |
|
CLIPTextEncode, |
|
) |
|
|
|
from modules.path import modelfile_path |
|
|
|
|
|
xl_base_filename = os.path.join(modelfile_path, 'sd_xl_base_1.0.safetensors') |
|
xl_refiner_filename = os.path.join(modelfile_path, 'sd_xl_refiner_1.0.safetensors') |
|
|
|
xl_base, xl_base_clip, xl_base_vae, xl_base_clipvision = load_checkpoint_guess_config(xl_base_filename) |
|
del xl_base_clipvision |
|
|
|
opCLIPTextEncode = CLIPTextEncode() |
|
opEmptyLatentImage = EmptyLatentImage() |
|
opKSamplerAdvanced = KSamplerAdvanced() |
|
|
|
|
|
positive_embedding = opCLIPTextEncode.encode(clip=xl_base_clip, text='a handsome man in forest') |
|
negative_embedding = opCLIPTextEncode.encode(clip=xl_base_clip, text='bad, ugly') |
|
|
|
initial_latent_image = opEmptyLatentImage.generate(width=1024, height=1536, batch_size=1) |
|
|
|
a = 0 |
|
|