Skip to content

Commit 75a818c

Browse files
Move mochi latent node to: latent/video.
1 parent 2865f91 commit 75a818c

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

comfy_extras/nodes_mochi.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -3,9 +3,6 @@
33
import comfy.model_management
44

55
class EmptyMochiLatentVideo:
6-
def __init__(self):
7-
self.device = comfy.model_management.intermediate_device()
8-
96
@classmethod
107
def INPUT_TYPES(s):
118
return {"required": { "width": ("INT", {"default": 848, "min": 16, "max": nodes.MAX_RESOLUTION, "step": 16}),
@@ -15,10 +12,10 @@ def INPUT_TYPES(s):
1512
RETURN_TYPES = ("LATENT",)
1613
FUNCTION = "generate"
1714

18-
CATEGORY = "latent/mochi"
15+
CATEGORY = "latent/video"
1916

2017
def generate(self, width, height, length, batch_size=1):
21-
latent = torch.zeros([batch_size, 12, ((length - 1) // 6) + 1, height // 8, width // 8], device=self.device)
18+
latent = torch.zeros([batch_size, 12, ((length - 1) // 6) + 1, height // 8, width // 8], device=comfy.model_management.intermediate_device())
2219
return ({"samples":latent}, )
2320

2421
NODE_CLASS_MAPPINGS = {

0 commit comments

Comments
 (0)