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

try fix colab (#377)

Browse Source
This commit is contained in:
lllyasviel 2023-09-14 21:37:29 -07:00 committed by GitHub
parent bcee7fd273
commit b5b4fd27f1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 17 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
fooocus_version.py
View File

@ -1 +1 @@
version = '2.0.12' version = '2.0.14'

19
modules/default_pipeline.py
View File

@ -2,6 +2,7 @@ import modules.core as core
import os import os
import torch import torch
import modules.path import modules.path
import comfy.model_management as model_management
from comfy.model_base import SDXL, SDXLRefiner from comfy.model_base import SDXL, SDXLRefiner
from modules.patch import cfg_patched from modules.patch import cfg_patched
@ -132,6 +133,8 @@ def clip_encode(sd, texts, pool_top_k=1):
if len(texts) == 0: if len(texts) == 0:
return None return None
model_management.soft_empty_cache()
clip = sd.clip clip = sd.clip
cond_list = [] cond_list = []
pooled_acc = 0 pooled_acc = 0
@ -162,16 +165,24 @@ def clear_all_caches():
@torch.no_grad() @torch.no_grad()
def process_diffusion(positive_cond, negative_cond, steps, switch, width, height, image_seed, callback): def patch_all_models():
if xl_base is not None: assert xl_base is not None
xl_base.unet.model_options['sampler_cfg_function'] = cfg_patched assert xl_base_patched is not None
if xl_base_patched is not None: xl_base.unet.model_options['sampler_cfg_function'] = cfg_patched
xl_base_patched.unet.model_options['sampler_cfg_function'] = cfg_patched xl_base_patched.unet.model_options['sampler_cfg_function'] = cfg_patched
if xl_refiner is not None: if xl_refiner is not None:
xl_refiner.unet.model_options['sampler_cfg_function'] = cfg_patched xl_refiner.unet.model_options['sampler_cfg_function'] = cfg_patched
return
@torch.no_grad()
def process_diffusion(positive_cond, negative_cond, steps, switch, width, height, image_seed, callback):
patch_all_models()
model_management.soft_empty_cache()
empty_latent = core.generate_empty_latent(width=width, height=height, batch_size=1) empty_latent = core.generate_empty_latent(width=width, height=height, batch_size=1)
if xl_refiner is not None: if xl_refiner is not None: