mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-08-06 21:28:06 +00:00
Fix another issue with the PR. (#9170)
This commit is contained in:
parent
84f9759424
commit
03895dea7c
@ -162,7 +162,7 @@ class BaseModel(torch.nn.Module):
|
||||
xc = self.model_sampling.calculate_input(sigma, x)
|
||||
|
||||
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
|
||||
dtype = self.get_dtype()
|
||||
|
Loading…
x
Reference in New Issue
Block a user