class
VJEPA2ACForWorldModeling
extends
WorldModelingModelVJEPA2ACForWorldModeling(config: VJEPA2ACConfig)V-JEPA 2-AC under the zoo's world-modeling task.
Parameters
configVJEPA2ACConfigFrozen model configuration.
Attributes
vjepa2_acVJEPA2ACModelThe encoder and the action-conditioned predictor. The wrapper
adds no parameters of its own.
Notes
It exists so AutoModelForWorldModeling can reach the family and so
the registry can tag it by task. The forward is one teacher-forced
latent transition; a rollout is that forward applied to its own
output, and the controller that decides the actions stays outside.
Used by 2
Constructors
1Instance methods
1forward(video: Tensor, actions: Tensor, states: Tensor, extrinsics: Tensor | None = None)Return an action-conditioned latent transition.