Skip to content

Commit 115f418

Browse files
Make EmptySD3LatentImage node use intermediate dtype. (Comfy-Org#13577)
1 parent 7385eb2 commit 115f418

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

comfy_extras/nodes_sd3.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ def define_schema(cls):
5454

5555
@classmethod
5656
def execute(cls, width, height, batch_size=1) -> io.NodeOutput:
57-
latent = torch.zeros([batch_size, 16, height // 8, width // 8], device=comfy.model_management.intermediate_device())
57+
latent = torch.zeros([batch_size, 16, height // 8, width // 8], device=comfy.model_management.intermediate_device(), dtype=comfy.model_management.intermediate_dtype())
5858
return io.NodeOutput({"samples": latent, "downscale_ratio_spacial": 8})
5959

6060
generate = execute # TODO: remove

0 commit comments

Comments
 (0)