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

Fix another issue with the PR. (#9170)

Browse Source
This commit is contained in:
comfyanonymous 2025-08-04 01:33:04 -07:00 committed by GitHub
parent 84f9759424
commit 03895dea7c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_base.py
View File

@ -162,7 +162,7 @@ class BaseModel(torch.nn.Module):
xc = self.model_sampling.calculate_input(sigma, x) xc = self.model_sampling.calculate_input(sigma, x)
if c_concat is not None: if c_concat is not None:
xc = torch.cat([xc] + [c_concat], dim=1) xc = torch.cat([xc] + [comfy.model_management.cast_to_device(c_concat, xc.device, xc.dtype)], dim=1)
context = c_crossattn context = c_crossattn
dtype = self.get_dtype() dtype = self.get_dtype()