alphard/efficiency-nodes-comfyui
1
0
Fork 0
mirror of https://github.com/jags111/efficiency-nodes-comfyui.git synced 2026-03-21 21:22: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_width = min(shape[-1] * 8, tile_width)
tile_height = min(shape[2] * 8, tile_height) tile_height = min(shape[2] * 8, tile_height)
real_model = None # real_model = None
# positive_copy = comfy.sample.convert_cond(positive) # positive_copy = comfy.sample.convert_cond(positive)
# negative_copy = comfy.sample.convert_cond(negative) # 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. #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) conds0 = \
negative_copy = comfy.sampler_helpers.convert_cond(negative) {"positive": comfy.sampler_helpers.convert_cond(positive),
modelPatches, inference_memory = comfy.sample.get_additional_models(positive_copy, negative_copy, model.model_dtype()) "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) comfy.model_management.load_models_gpu([model] + modelPatches, model.memory_required(noise.shape) + inference_memory)
real_model = model.model real_model = model.model
#smaller changes #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): for m, img in zip(T2Is, T2I_imgs):
slices_T2I(tile_h, tile_h_len, tile_w, tile_w_len, m, img) 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) pos = [c.copy() for c in positive]#copy_cond(positive_copy)
neg = [c.copy() for c in negative_copy]#copy_cond(negative_copy) neg = [c.copy() for c in negative]#copy_cond(negative_copy)
#cond areas #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)] 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() samples = samples_next.clone()
comfy.sample.cleanup_additional_models(modelPatches) comfy.sampler_helpers.cleanup_additional_models(modelPatches)
out = latent_image.copy() out = latent_image.copy()
out["samples"] = samples.cpu() out["samples"] = samples.cpu()