HazelNam

/

CJEPA

like 5

• "torch.FloatStorage",
• "torch.float32",
• "videosaur.videosaur.modules.networks.Attention",
• "torch.nn.functional.relu",
• "torch.nn.modules.dropout.Dropout",
• "torch.nn.modules.activation.ReLU",
• "transformers.models.dinov2.modeling_dinov2.Dinov2Embeddings",
• "transformers.models.dinov2.modeling_dinov2.Dinov2PatchEmbeddings",
• "transformers.models.dinov2.modeling_dinov2.Dinov2SelfOutput",
• "omegaconf.dictconfig.DictConfig",
• "lightning.fabric.utilities.data.AttributeDict",
• "videosaur.videosaur.modules.video.LatentProcessor",
• "custom_models.cjepa_predictor.MaskedSlot_AP_Predictor",
• "omegaconf.nodes.AnyNode",
• "__builtin__.list",
• "__builtin__.long",
• "torch.nn.modules.container.ModuleList",
• "torch.device",
• "torch.nn.modules.conv.Conv1d",
• "torch.nn.modules.container.Sequential",
• "torch.nn.modules.activation.MultiheadAttention",
• "__builtin__.getattr",
• "__builtin__.set",
• "torch._utils._rebuild_parameter",
• "videosaur.videosaur.modules.networks.TransformerEncoderLayer",
• "torch.nn.modules.normalization.LayerNorm",
• "torch.nn.modules.activation.GELU",
• "transformers.models.dinov2.configuration_dinov2.Dinov2Config",
• "torch.nn.modules.rnn.GRUCell",
• "videosaur.videosaur.modules.video.MapOverTime",
• "videosaur.videosaur.modules.networks.MLP",
• "custom_models.dinowm_causal_AP_node.CausalWM_AP",
• "transformers.models.dinov2.modeling_dinov2.Dinov2Encoder",
• "__builtin__.dict",
• "videosaur.videosaur.modules.encoders.FrameEncoder",
• "torch.nn.modules.linear.Linear",
• "transformers.models.dinov2.modeling_dinov2.Dinov2Layer",
• "collections.OrderedDict",
• "torchvision.transforms.v2._container.Compose",
• "torch.nn.modules.linear.Identity",
• "transformers.activations.GELUActivation",
• "stable_worldmodel.wm.dinowm.Embedder",
• "torch.nn.modules.container.ModuleDict",
• "stable_pretraining.backbone.utils.EvalOnly",
• "transformers.models.dinov2.modeling_dinov2.Dinov2SelfAttention",
• "collections.defaultdict",
• "torch._utils._rebuild_tensor_v2",
• "torch.nn.modules.linear.NonDynamicallyQuantizableLinear",
• "transformers.models.dinov2.modeling_dinov2.Dinov2Attention",
• "typing.Any",
• "torch._C._nn.gelu",
• "videosaur.videosaur.modules.video.ScanOverTime",
• "videosaur.videosaur.modules.networks.TransformerEncoder",
• "videosaur.videosaur.modules.initializers.RandomInit",
• "stable_pretraining.module.Module",
• "torch.nn.modules.conv.Conv2d",
• "transformers.models.dinov2.modeling_dinov2.Dinov2MLP",
• "transformers.models.dinov2.modeling_dinov2.Dinov2Model",
• "omegaconf.base.ContainerMetadata",
• "omegaconf.base.Metadata",
• "transformers.models.dinov2.modeling_dinov2.Dinov2LayerScale",
• "torchvision.transforms.functional.InterpolationMode",
• "videosaur.videosaur.modules.groupers.SlotAttention",
• "omegaconf.listconfig.ListConfig",
• "custom_models.cjepa_predictor.NonCausalTransformer",
• "torchvision.transforms.v2._geometry.Resize"

How to fix it?