Skip to content

Commit 03895de

Browse files
Fix another issue with the PR. (Comfy-Org#9170)
1 parent 84f9759 commit 03895de

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

comfy/model_base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -162,7 +162,7 @@ def _apply_model(self, x, t, c_concat=None, c_crossattn=None, control=None, tran
162162
xc = self.model_sampling.calculate_input(sigma, x)
163163

164164
if c_concat is not None:
165-
xc = torch.cat([xc] + [c_concat], dim=1)
165+
xc = torch.cat([xc] + [comfy.model_management.cast_to_device(c_concat, xc.device, xc.dtype)], dim=1)
166166

167167
context = c_crossattn
168168
dtype = self.get_dtype()

0 commit comments

Comments
 (0)