From f3323324951772cc15f0e5fd8aaccae03853968a Mon Sep 17 00:00:00 2001 From: Haoming Date: Thu, 11 Dec 2025 10:40:11 +0800 Subject: [PATCH] device --- comfy/ldm/lumina/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/ldm/lumina/model.py b/comfy/ldm/lumina/model.py index 123c191a9..0ff8539f7 100644 --- a/comfy/ldm/lumina/model.py +++ b/comfy/ldm/lumina/model.py @@ -628,7 +628,7 @@ class NextDiT(nn.Module): transformer_options["total_blocks"] = len(self.layers) transformer_options["block_type"] = "double" for i, layer in enumerate(self.layers): - transformer_options["block_index"] = torch.tensor(i, dtype=torch.uint8, device=img.device) + transformer_options["block_index"] = torch.tensor(i, dtype=torch.uint8, device="cpu") img = layer(img, mask, freqs_cis, adaln_input, transformer_options=transformer_options) if "double_block" in patches: for p in patches["double_block"]: