Skip to content

Commit e0010da

Browse files
comfyanonymousadlerfaulkner
authored andcommitted
Temp fix for LTXV custom nodes. (comfyanonymous#10251)
1 parent 9d27b7e commit e0010da

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

comfy_extras/nodes_lt.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@ def execute(cls, width, height, length, batch_size=1) -> io.NodeOutput:
3434
latent = torch.zeros([batch_size, 128, ((length - 1) // 8) + 1, height // 32, width // 32], device=comfy.model_management.intermediate_device())
3535
return io.NodeOutput({"samples": latent})
3636

37+
generate = execute # TODO: remove
3738

3839
class LTXVImgToVideo(io.ComfyNode):
3940
@classmethod
@@ -77,6 +78,8 @@ def execute(cls, positive, negative, image, vae, width, height, length, batch_si
7778

7879
return io.NodeOutput(positive, negative, {"samples": latent, "noise_mask": conditioning_latent_frames_mask})
7980

81+
generate = execute # TODO: remove
82+
8083

8184
def conditioning_get_any_value(conditioning, key, default=None):
8285
for t in conditioning:
@@ -264,6 +267,8 @@ def execute(cls, positive, negative, vae, latent, image, frame_idx, strength) ->
264267

265268
return io.NodeOutput(positive, negative, {"samples": latent_image, "noise_mask": noise_mask})
266269

270+
generate = execute # TODO: remove
271+
267272

268273
class LTXVCropGuides(io.ComfyNode):
269274
@classmethod
@@ -300,6 +305,8 @@ def execute(cls, positive, negative, latent) -> io.NodeOutput:
300305

301306
return io.NodeOutput(positive, negative, {"samples": latent_image, "noise_mask": noise_mask})
302307

308+
crop = execute # TODO: remove
309+
303310

304311
class LTXVConditioning(io.ComfyNode):
305312
@classmethod
@@ -498,6 +505,7 @@ def execute(cls, image, img_compression) -> io.NodeOutput:
498505
output_images.append(preprocess(image[i], img_compression))
499506
return io.NodeOutput(torch.stack(output_images))
500507

508+
preprocess = execute # TODO: remove
501509

502510
class LtxvExtension(ComfyExtension):
503511
@override

0 commit comments

Comments
 (0)