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

Make sure context is on the right device. (#9154)

Browse Source
This commit is contained in:
comfyanonymous 2025-08-02 12:09:23 -07:00 committed by GitHub
parent 5f582a9757
commit 13aaa66ec2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/model_base.py
View File

@ -171,12 +171,12 @@ class BaseModel(torch.nn.Module):
dtype = self.manual_cast_dtype dtype = self.manual_cast_dtype
xc = xc.to(dtype) xc = xc.to(dtype)
device = xc.device
t = self.model_sampling.timestep(t).float() t = self.model_sampling.timestep(t).float()
if context is not None: if context is not None:
context = context.to(dtype) context = context.to(dtype=dtype, device=device)
extra_conds = {} extra_conds = {}
device = xc.device
for o in kwargs: for o in kwargs:
extra = kwargs[o] extra = kwargs[o]