dreamdifferent

/

mimic-video-so101-multi-object-delta30-action-decoder

like 0

Follow

dreamdifferent 3

• "imaginaire.trainer.ImaginaireTrainer",
• "cosmos_predict2.callbacks.grad_clip.GradClip",
• "cosmos_predict2.configs.config.Config",
• "imaginaire.config.Profiling",
• "imaginaire.utils.callback.ProgressBarCallback",
• "imaginaire.auxiliary.text_encoder.CosmosT5TextEncoderConfig",
• "omegaconf.nodes.StringNode",
• "omegaconf.base.Metadata",
• "cosmos_predict2.configs.config_world2action.SchedulerConfig",
• "cosmos_predict2.models.world2action_model.World2ActionModel",
• "cosmos_predict2.data.resumable_sampler.ResumableDistributedSampler",
• "builtins.list",
• "builtins.str",
• "cosmos_predict2.functional.lr_scheduler.LambdaLinearScheduler",
• "cosmos_predict2.models.text2image_dit.SACConfig",
• "omegaconf.nodes.EnumNode",
• "builtins.float",
• "builtins.bool",
• "cosmos_predict2.configs.defaults.ema.EMAConfig",
• "cosmos_predict2.configs.defaults.data_action.get_dataset",
• "builtins.int",
• "cosmos_predict2.configs.experiment.world2action.get_local_batch_size",
• "collections.defaultdict",
• "megatron.core.parallel_state.get_data_parallel_rank",
• "cosmos_predict2.callbacks.device_monitor.DeviceMonitor",
• "cosmos_predict2.checkpointer.Checkpointer",
• "imaginaire.config.TrainerConfig",
• "omegaconf.base.ContainerMetadata",
• "megatron.core.parallel_state.get_data_parallel_world_size",
• "cosmos_predict2.utils.optim_instantiate_dtensor.get_base_optimizer",
• "cosmos_predict2.configs.config_video2world.CosmosGuardrailConfig",
• "imaginaire.utils.callback.EMAModelCallback",
• "cosmos_predict2.configs.defaults.data_action.get_data_config",
• "torch._utils.torch.preserve_format",
• "omegaconf.dictconfig.DictConfig",
• "omegaconf.nodes.AnyNode",
• "cosmos_predict2.conditioner.VideoConditioner",
• "imaginaire.config.DDPConfig",
• "megatron.core.model_parallel_config.ModelParallelConfig",
• "builtins.dict",
• "imaginaire.callbacks.manual_gc.ManualGarbageCollection",
• "imaginaire.utils.callback.WandbCallback",
• "imaginaire.config.CuDNNConfig",
• "cosmos_predict2.configs.config_video2world.SolverTimestampConfig",
• "omegaconf.nodes.IntegerNode",
• "torch.utils.data.dataloader.DataLoader",
• "cosmos_predict2.models.world2action_dit.World2ActionDIT",
• "cosmos_predict2.tokenizers.tokenizer.TokenizerInterface",
• "omegaconf.nodes.FloatNode",
• "cosmos_predict2.configs.config_video2world.Video2WorldPipelineConfig",
• "torch.float32",
• "imaginaire.config.CheckpointConfig",
• "typing.List",
• "imaginaire.utils.callback.LowPrecisionCallback",
• "cosmos_predict2.conditioner.ReMapkey",
• "omegaconf.nodes.BooleanNode",
• "cosmos_predict2.models.video2world_dit.MinimalV1LVGDiT",
• "cosmos_predict2.callbacks.iter_speed.IterSpeed",
• "imaginaire.auxiliary.text_encoder.CosmosTextEncoderConfig",
• "cosmos_predict2.conditioner.BooleanFlag",
• "imaginaire.constants.TextEncoderClass",
• "imaginaire.config.JITConfig",
• "cosmos_predict2.conditioner.TextAttr",
• "omegaconf.listconfig.ListConfig",
• "imaginaire.config.JobConfig",
• "typing.Any"

How to fix it?