nodes/BlenderNeko-ComfyUI_TiledKSampler
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

don't accidentally break normal f16 cnets

Browse Source
This commit is contained in:
BlenderNeko
2023-12-25 22:44:21 +01:00
parent 1c1ecba9f5
commit 4fcc003f62
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+5 -1
View File
@@ -83,7 +83,11 @@ def slice_gligen(tile_h, tile_h_len, tile_w, tile_w_len, cond, gligen):
def slice_cnet(h, h_len, w, w_len, model:comfy.controlnet.ControlBase, img):
if img is None:
img = model.cond_hint_original
model.cond_hint = tiling.get_slice(img, h*8, h_len*8, w*8, w_len*8).float().to(model.device)
hint = tiling.get_slice(img, h*8, h_len*8, w*8, w_len*8)
if isinstance(model, comfy.controlnet.ControlLora):
model.cond_hint = hint.float().to(model.device)
else:
model.cond_hint = hint.to(model.control_model.dtype).to(model.device)
def slices_T2I(h, h_len, w, w_len, model:comfy.controlnet.ControlBase, img):
model.control_input = None