mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-02-15 07:52:35 +08:00
forgot about ruff
This commit is contained in:
parent
07cd971992
commit
291b091720
@ -113,7 +113,7 @@ class EncodeVideoVAE(io.ComfyNode):
|
|||||||
model = vae.first_stage_model
|
model = vae.first_stage_model
|
||||||
model = model.to(vae.device)
|
model = model.to(vae.device)
|
||||||
return io.NodeOutput(encode_video(vae, model, video, step_size, processing_batch_size))
|
return io.NodeOutput(encode_video(vae, model, video, step_size, processing_batch_size))
|
||||||
|
|
||||||
class EncodeVideoCLIP(io.ComfyNode):
|
class EncodeVideoCLIP(io.ComfyNode):
|
||||||
@classmethod
|
@classmethod
|
||||||
def define_schema(cls):
|
def define_schema(cls):
|
||||||
@ -130,7 +130,7 @@ class EncodeVideoCLIP(io.ComfyNode):
|
|||||||
io.Conditioning.Output(display_name="encoded_video"),
|
io.Conditioning.Output(display_name="encoded_video"),
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def execute(cls, video, processing_batch_size, step_size, clip_vision):
|
def execute(cls, video, processing_batch_size, step_size, clip_vision):
|
||||||
model = clip_vision.model
|
model = clip_vision.model
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user