Update bnk_tiled_samplers.py

This commit is contained in:
VALADI K JAGANATHAN
2023-11-01 14:08:55 +05:30
committed by GitHub
parent 08842dc8f1
commit 4071b0799d

View File

@@ -135,7 +135,7 @@ def sample_common(model, add_noise, noise_seed, tile_width, tile_height, tiling_
samples += sampler.sigmas[start_at_step].cpu() * model.model.process_latent_out(noise) samples += sampler.sigmas[start_at_step].cpu() * model.model.process_latent_out(noise)
#cnets #cnets
cnets = [c['control'] for (_, c) in positive + negative if 'control' in c and isinstance(c['control'], comfy.controlnet.ControlNet)] cnets = [c['control'] for (_, c) in positive + negative if 'control' in c and isinstance(c['control'], comfy.controlnet.ControlNet)]
cnets = list(set([x for m in cnets for x in recursion_to_list(m, "previous_controlnet")])) cnets = list(set([x for m in cnets for x in recursion_to_list(m, "previous_controlnet")]))
cnet_imgs = [ cnet_imgs = [
torch.nn.functional.interpolate(m.cond_hint_original, (shape[-2] * 8, shape[-1] * 8), mode='nearest-exact').to('cpu') torch.nn.functional.interpolate(m.cond_hint_original, (shape[-2] * 8, shape[-1] * 8), mode='nearest-exact').to('cpu')
@@ -284,7 +284,7 @@ def sample_common(model, add_noise, noise_seed, tile_width, tile_height, tiling_
out["samples"] = samples.cpu() out["samples"] = samples.cpu()
return (out, ) return (out, )
+
class TiledKSampler: class TiledKSampler:
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):