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 
memory management changed in comfy original repo so the code got changed.
This commit is contained in:
VALADI K JAGANATHAN
2023-11-26 12:27:01 +05:30
committed by GitHub
parent 94d07c3154
commit a077975c3e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
py/bnk_tiled_samplers.py
View File

@@ -123,7 +123,7 @@ def sample_common(model, add_noise, noise_seed, tile_width, tile_height, tiling_
positive_copy = comfy.sample.convert_cond(positive)
negative_copy = comfy.sample.convert_cond(negative)
modelPatches, inference_memory = comfy.sample.get_additional_models(positive_copy, negative_copy, model.model_dtype())
comfy.model_management.load_models_gpu([model] + modelPatches, comfy.model_management.batch_area_memory(noise.shape[0] * noise.shape[2] * noise.shape[3]) + inference_memory)
comfy.model_management.load_models_gpu([model] + modelPatches, memory_required=(noise.shape[0] * noise.shape[2] * noise.shape[3]) + inference_memory)
real_model = model.model
sampler = comfy.samplers.KSampler(real_model, steps=steps, device=device, sampler=sampler_name, scheduler=scheduler, denoise=denoise, model_options=model.model_options)