mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-06-08 07:07:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make the casting in lists the same as regular inputs. (#8373)

Browse Source
This commit is contained in:
comfyanonymous 2025-06-01 02:39:54 -07:00 committed by GitHub
parent 180db6753f
commit fb4754624d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
comfy/model_base.py
View File

@ -102,6 +102,13 @@ def model_sampling(model_config, model_type):
return ModelSampling(model_config) return ModelSampling(model_config)
def convert_tensor(extra, dtype):
if hasattr(extra, "dtype"):
if extra.dtype != torch.int and extra.dtype != torch.long:
extra = extra.to(dtype)
return extra
class BaseModel(torch.nn.Module): class BaseModel(torch.nn.Module):
def __init__(self, model_config, model_type=ModelType.EPS, device=None, unet_model=UNetModel): def __init__(self, model_config, model_type=ModelType.EPS, device=None, unet_model=UNetModel):
super().__init__() super().__init__()
@ -165,13 +172,13 @@ class BaseModel(torch.nn.Module):
extra_conds = {} extra_conds = {}
for o in kwargs: for o in kwargs:
extra = kwargs[o] extra = kwargs[o]
if hasattr(extra, "dtype"): if hasattr(extra, "dtype"):
if extra.dtype != torch.int and extra.dtype != torch.long: extra = convert_tensor(extra, dtype)
extra = extra.to(dtype) elif isinstance(extra, list):
if isinstance(extra, list):
ex = [] ex = []
for ext in extra: for ext in extra:
ex.append(ext.to(dtype)) ex.append(convert_tensor(ext, dtype))
extra = ex extra = ex
extra_conds[o] = extra extra_conds[o] = extra