dmarx / ComfyUI-Keyframed

ComfyUI nodes to facilitate prompt/parameter keyframing
MIT License
86 stars 7 forks source link

Keyframed Nodes For ComfyUI

ComfyUI nodes to facilitate parameter/prompt keyframing using comfyui nodes for defining and manipulating parameter curves. Essentially provides a ComfyUI interface to the keyframed library.

Related project: https://github.com/FizzleDorf/ComfyUI_FizzNodes


Overview

The nodes in this extension support parameterizing animations whose prompts or other settings will change over time. Other systems for achieving this currently exist in the ComfyUI and AI art ecosystem which rely heavily on notation. The motivation of this extension is to take full advantage of ComfyUI's node system for manipulating "keyframed" parameters like this, hopefully making this kind of work accessible to people who may otherwise be intimidated or constrained by the notation systems currently in use.

The nodes here are mostly wrappers around the python library keyframed. The main objects we inherit from keyframed are:

These nodes introduce two additional objects to facilitate working with prompts

Starter Workflows

Prompt Scheduling

see also: Nodes > Scheduling

Prompt Scheduling

This one is probably why you are here. This workflow demonstrates how to use the keyframed/schedule nodes to achieve similar behavior as FizzNodes' PromptSchedule node, but implemented differently.

This schedule is essentailly a normal AnimateDiff workflow where several nodes have replaced the normal conditioning setup. Rather than a single CLIP Text Encode node, we can have multiple prompts which transition sequentially over time.

Here's a more fully worked version of the workflow above (adds a second refinement pass of AnimateDiff, FiLM VFI, and upscaling).

Prompt Scheduling

These workflows illustrate the most user-friendly prompt-scheduling interface, but there are "lower-level" nodes as well for more complex bespoke scheduling.

Prompt Scheduling

Interleaving Multiple Prompts Simultaneously (aka Prompt Entanglement, aka Prompt Superposition)

Prompt Entanglement

Workflow output: https://twitter.com/DigThatData/status/1733416414864957484

Workflow features:

The "Nx Entangled Curves" nodes generate N sinusoidal curves whose phases are offset to be equally spaced and whose amplitudes are adjusted such that the sum of all curves at any time is 1.

Prompt Entanglement

This can be used to create a "prompt superposition" effect, where all prompts contribute at any given time, and their relative strengths varies throughout the animation, each taking turns being the strongest or weakest contributor to the conditioning. This differs from the "prompt scheduling" workflow above, where at most two prompts are active at any given time (i.e. during transitions between keyframes).

For documentation detailing how this workflow works, see the Nodes > Entangled Curves section below.

Simple Curved Parameter

Simple Curved Parameter

In this example we ilustrate how arithmetic can be performed on curves. Curves also support arithmetic operations against floats and ints, e.g. myCurve * 3.

Multi-Prompt Transition With Manually Specified Curves

Manual Prompt Transition

If you're feeling adventurous, this workflow demonstrates how you would use the curve objects directly to acheive the same thing as a schedule. Each prompt gets its own seaparate curve indicating the weight of the prompt at that time (you probably want the various conditionings weights to sum to 1 when combined. If you need to "fill" missing conditioning weight, try using an empty prompt).

Parameter Groups and Curve Drawing Utilities

Parameter Groups and Curve Drawing Utilities

Parameter Groups can be used to carry around multiple curves around. If you want to attach a curve to a parameter group, it will need a label.

We provide utilities for drawing curves. You can draw multiple curves simultaneously by collecting them in a parameter group. The labels you give the curves will also be used as labels for plotting. You can also plot labels you haven't curves (a label was randomly generated when the curve was initialized).

Nodes

Curve Constructors

Curve From String

Curve From String

Supports commonly used notation for curved parameter. See also Deforum, FizzNodes, https://www.chigozie.co.uk/keyframe-string-generator/

Curve From YAML

Curve From YAML

Supports curved parameter notation used by https://github.com/dmarx/keyframed

Constant-Valued Curve

Curve From YAML

Returns a curve which evaluates to the same value for all values of t (time). Especially useful for curve arithmetic.

Entangled Curves

Entangled Curves

Each output curve of the node is a sine wave that oscillates from 0 to 1 at the given frequency or wavelength. The outputs of a given node are phase-offset such that at any given time, the sum of the generated curves is 1.

Reference the Prompt Interleaving Workflow for a demonstrative example.

Curve Operators

Evaluate Curve At T

Evaluate Curve At T

Apply Curve To Conditioning

Apply Curve To Conditioning

Generates a batch of n conditionings multiplying each conditioning by th value of the curve at that conditioning's index within the batch. If a batch of latents is provided, n is ignored and the size of the latent batch is used. If the provided conditioning object has batch size > 1, the conditioning's batch size takes priority over the latent batch size and n.

Add Conditions

Add Conditions

Add Conditions (x10)

If you're using the x10 node, at least curve_0 must be non-empty. The other cond positions are all optionally populated.

Curve Arithmetic Operators

Curve Arithmetic

Arithmetic is performed at the union of keyframes of the provided curves.

NB: the division operator is unreliable at the time of this writing (2023-12-09).

Curve Arithmetic - batch pooling

If you have lots of curve objects to multiply together or add together, here are some convenience nodes.

Scheduling

These nodes work together to facilitate transitioning through a sequence of conditionings (i.e. prompts). We'll call this sequence the "schedule" of the conditionings. The primary use case here is for manipulating the positive prompt, i.e. for building a "prompt schedule". Given a particular time (e.g. frame id) in an animation sequence, we can query the prompt schedule at that time to get the appropriate conditioning to pass to the KSampler.

Reference the Prompt Scheduling Workflow for a demonstrative example

Schedule Prompt

Schedule Prompt

This will be the "workhorse" node provided by this package for the majority of people. It combines the nodes CLIPTextEncode, Keyframed Condition, and Set Keyframe (the latter two described below). It offers three outputs, but you will generally only need the SCHEDULE output. If an input schedule is not provided, a new SCHEDULE object will be created and provided at the output, which you can then pass to subsequent Schedule Prompt nodes to add keyframed conditions to the same schedule.

Keyframed Condition

Keyframed Condition

This node attaches a conditioning to a keyframe. This let's us assign a time to the conditioning and set what interpolation method to use when we're between keyframes.

Interpolation Methods

Consider three time points a,b,c such that a<b<c, and two keyframes X,Y such that X.time = a and Y.time = c. To interpolate a value at time b, we would use X.interpolation_method to "tween" the value between X.value and Y.value.

Set Keyframe

Set Keyframe

Attaches a keyframe to a schedule. If you haven't created a schedule yet, pass your keyframe into this node to create one, then pass the output schedule to subsequent Set Keyframe nodes to attach additional keyframes to the schedule. The first keyframe in the schedule should always be at time=0.

Evaluate Schedule

Evaluate Schedule

Evaluate the schedule at a time or time slice to extract a single conditioning or conditioning batch (respectively) to provide to e.g. a KSampler. The non-batch version gives the same kind of conditioning you'd get from a CLIPTextConditioning node. The intention of the batch version is to use with AnimateDiff to apply different conditionings per frame.