diff --git a/comfy/samplers.py b/comfy/samplers.py index f3064000..98b1932f 100644 --- a/comfy/samplers.py +++ b/comfy/samplers.py @@ -143,7 +143,7 @@ def cond_cat(c_list, device=None): for k in temp: conds = temp[k] out[k] = conds[0].concat(conds[1:]) - if device is not None: + if device is not None and hasattr(out[k], 'to'): out[k] = out[k].to(device) return out