mirrors/Fooocus
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

i

Browse Source
This commit is contained in:
lvmin 2023-08-10 05:37:25 -07:00
parent 683128e5f2
commit 7c576e8c67
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
webui.py
View File

@ -2,6 +2,7 @@ import os
import random import random
from comfy.sd import load_checkpoint_guess_config from comfy.sd import load_checkpoint_guess_config
from comfy.model_management import unload_model
from nodes import ( from nodes import (
VAEDecode, VAEDecode,
@ -26,6 +27,7 @@ opVAEDecode = VAEDecode()
positive_conditions = opCLIPTextEncode.encode(clip=xl_base_clip, text='a handsome man in forest')[0] positive_conditions = opCLIPTextEncode.encode(clip=xl_base_clip, text='a handsome man in forest')[0]
negative_conditions = opCLIPTextEncode.encode(clip=xl_base_clip, text='bad, ugly')[0] negative_conditions = opCLIPTextEncode.encode(clip=xl_base_clip, text='bad, ugly')[0]
unload_model()
initial_latent_image = opEmptyLatentImage.generate(width=1024, height=1024, batch_size=1)[0] initial_latent_image = opEmptyLatentImage.generate(width=1024, height=1024, batch_size=1)[0]
@ -44,6 +46,7 @@ samples = opKSamplerAdvanced.sample(
negative=negative_conditions, negative=negative_conditions,
latent_image=initial_latent_image, latent_image=initial_latent_image,
)[0] )[0]
unload_model()
vae_decoded = opVAEDecode.decode(samples=samples, vae=xl_base_vae)[0] vae_decoded = opVAEDecode.decode(samples=samples, vae=xl_base_vae)[0]