JeffrinSam

/

cosmoss01_lora

like 0

• "functools.partial",
• "cosmos_predict2._src.predict2.text_encoders.text_encoder.TextEncoderConfig",
• "cosmos_predict2._src.predict2.networks.minimal_v4_dit.SACConfig",
• "attr._make._Nothing",
• "builtins.int",
• "builtins.bool",
• "cosmos_predict2._src.predict2.callbacks.grad_clip.GradClip",
• "torch.float32",
• "cosmos_predict2._src.predict2.callbacks.heart_beat.HeartBeat",
• "cosmos_predict2._src.predict2.datasets.cached_replay_dataloader.duplicate_batches",
• "omegaconf.nodes.IntegerNode",
• "cosmos_predict2._src.predict2.datasets.local_datasets.dataset_video.VideoDataset",
• "cosmos_predict2._src.predict2.conditioner.BooleanFlag",
• "cosmos_predict2._src.predict2.configs.video2world.config.Config",
• "cosmos_predict2._src.reason1.configs.default.model_config.CommConfig",
• "cosmos_predict2._src.predict2.datasets.cached_replay_dataloader.get_cached_replay_dataloader",
• "cosmos_predict2._src.reason1.configs.default.model_config.VisionEncoderConfig",
• "cosmos_predict2._src.reason1.configs.default.model_config.OptimizerConfig",
• "builtins.list",
• "cosmos_predict2._src.reason1.configs.default.model_config.TrainingConfig",
• "collections.defaultdict",
• "cosmos_predict2._src.imaginaire.config.TrainerConfig",
• "megatron.core.model_parallel_config.ModelParallelConfig",
• "omegaconf.dictconfig.DictConfig",
• "cosmos_predict2._src.imaginaire.config.JITConfig",
• "typing.Any",
• "cosmos_predict2._src.reason1.configs.default.model_config.Float8Config",
• "torch._utils.torch.preserve_format",
• "cosmos_predict2._src.predict2.tokenizers.wan2pt1.Wan2pt1VAEInterface",
• "omegaconf.listconfig.ListConfig",
• "cosmos_predict2._src.predict2.conditioner.ReMapkey",
• "builtins.float",
• "cosmos_predict2._src.predict2.conditioner.TextAttr",
• "cosmos_predict2._src.imaginaire.config.ObjectStoreConfig",
• "cosmos_predict2._src.predict2.datasets.cached_replay_dataloader.duplicate_batches_random",
• "cosmos_predict2._src.imaginaire.config.StragglerDetectionConfig",
• "builtins.str",
• "cosmos_predict2._src.predict2.callbacks.wandb_log.WandbCallback",
• "omegaconf.base.ContainerMetadata",
• "cosmos_predict2._src.predict2.callbacks.dataloading_monitor.DetailedDataLoadingSpeedMonitor",
• "cosmos_predict2._src.predict2.callbacks.compile_tokenizer.CompileTokenizer",
• "cosmos_predict2._src.imaginaire.config.CheckpointConfig",
• "cosmos_predict2._src.imaginaire.config.DDPConfig",
• "cosmos_predict2._src.predict2.callbacks.iter_speed.IterSpeed",
• "cosmos_predict2._src.imaginaire.utils.callback.LowPrecisionCallback",
• "typing.Dict",
• "cosmos_predict2._src.predict2.callbacks.device_monitor.DeviceMonitor",
• "omegaconf.nodes.BooleanNode",
• "cosmos_predict2._src.predict2.datasets.data_sources.mock_data.get_image_dataset",
• "cosmos_predict2._src.predict2.datasets.local_datasets.dataset_video.get_generic_dataloader",
• "cosmos_predict2._src.predict2.models.video2world_model_rectified_flow.Video2WorldModelRectifiedFlow",
• "omegaconf.nodes.AnyNode",
• "cosmos_predict2._src.imaginaire.config.JobConfig",
• "cosmos_predict2._src.predict2.datasets.local_datasets.dataset_video.get_sampler",
• "cosmos_predict2._src.predict2.models.video2world_model_rectified_flow.Video2WorldModelRectifiedFlowConfig",
• "cosmos_predict2._src.reason1.configs.default.model_config.ActivationCheckpointConfig",
• "typing.List",
• "cosmos_predict2._src.predict2.checkpointer.dcp.DistributedCheckpointer",
• "cosmos_predict2._src.predict2.networks.minimal_v1_lvg_dit.MinimalV1LVGDiT",
• "omegaconf.base.Metadata",
• "omegaconf.nodes.FloatNode",
• "cosmos_predict2._src.imaginaire.config.CuDNNConfig",
• "cosmos_predict2._src.predict2.callbacks.every_n_draw_sample.EveryNDrawSample",
• "builtins.dict",
• "cosmos_predict2._src.common.functional.lr_scheduler.LambdaLinearScheduler",
• "cosmos_predict2._src.imaginaire.callbacks.manual_gc.ManualGarbageCollection",
• "cosmos_predict2._src.predict2.datasets.joint_dataloader.IterativeJointDataLoader",
• "cosmos_predict2._src.predict2.configs.video2world.defaults.conditioner.Video2WorldConditioner",
• "cosmos_predict2._src.reason1.configs.default.model_config.CheckpointConfig",
• "omegaconf.nodes.StringNode",
• "cosmos_predict2._src.imaginaire.trainer.ImaginaireTrainer",
• "cosmos_predict2._src.predict2.utils.optim_instantiate.get_base_optimizer",
• "cosmos_predict2._src.reason1.configs.default.model_config.ExperimentalConfig",
• "cosmos_predict2._src.predict2.models.text2world_model.EMAConfig",
• "cosmos_predict2._src.reason1.tokenizer.processor.build_tokenizer",
• "cosmos_predict2._src.predict2.datasets.data_sources.mock_data.get_video_dataset",
• "cosmos_predict2._src.predict2.text_encoders.reason1.QwenVLBaseModel",
• "cosmos_predict2._src.imaginaire.config.Profiling"

How to fix it?