alphard/efficiency-nodes-comfyui
1
0
Fork 0
mirror of https://github.com/jags111/efficiency-nodes-comfyui.git synced 2026-03-21 13:12:13 -03:00
Code Issues Packages Projects Releases Wiki Activity

Update bnk_tiled_samplers.py

Browse Source
This commit is contained in:
VALADI K JAGANATHAN
2024-05-31 18:05:44 +05:30
committed by GitHub
parent c5773181ed
commit 39c6758198
1 changed files with 13 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
py/bnk_tiled_samplers.py
View File

@@ -119,13 +119,19 @@ def sample_common(model, add_noise, noise_seed, tile_width, tile_height, tiling_
tile_width = min(shape[-1] * 8, tile_width)
tile_height = min(shape[2] * 8, tile_height)
real_model = None
# real_model = None
# positive_copy = comfy.sample.convert_cond(positive)
# negative_copy = comfy.sample.convert_cond(negative)
#change made due to changes in omfy.sampler's 'convert cond' has been removed.. check if any error persists.
positive_copy = comfy.sampler_helpers.convert_cond(positive)
negative_copy = comfy.sampler_helpers.convert_cond(negative)
modelPatches, inference_memory = comfy.sample.get_additional_models(positive_copy, negative_copy, model.model_dtype())
conds0 = \
{"positive": comfy.sampler_helpers.convert_cond(positive),
"negative": comfy.sampler_helpers.convert_cond(negative)}
conds = {}
for k in conds0:
conds[k] = list(map(lambda a: a.copy(), conds0[k]))
modelPatches, inference_memory = comfy.sampler_helpers.get_additional_models(conds, model.model_dtype())
comfy.model_management.load_models_gpu([model] + modelPatches, model.memory_required(noise.shape) + inference_memory)
real_model = model.model
#smaller changes
@@ -256,8 +262,8 @@ def sample_common(model, add_noise, noise_seed, tile_width, tile_height, tiling_
for m, img in zip(T2Is, T2I_imgs):
slices_T2I(tile_h, tile_h_len, tile_w, tile_w_len, m, img)
pos = [c.copy() for c in positive_copy]#copy_cond(positive_copy)
neg = [c.copy() for c in negative_copy]#copy_cond(negative_copy)
pos = [c.copy() for c in positive]#copy_cond(positive_copy)
neg = [c.copy() for c in negative]#copy_cond(negative_copy)
#cond areas
pos = [slice_cond(tile_h, tile_h_len, tile_w, tile_w_len, c, area, device) for c, area in zip(pos, spatial_conds_pos)]
@@ -283,7 +289,7 @@ def sample_common(model, add_noise, noise_seed, tile_width, tile_height, tiling_
samples = samples_next.clone()
comfy.sample.cleanup_additional_models(modelPatches)
comfy.sampler_helpers.cleanup_additional_models(modelPatches)
out = latent_image.copy()
out["samples"] = samples.cpu()