surgical-cosmos / cosmos_predict2_2b_480p_knot_tying

19.6 GB

Ctrl+K

1 contributor

History: 3 commits

This model has 24 files scanned as unsafe.

ruirui4

Add files using upload-large-folder tool

ecadaea verified 5 days ago

checkpoints
Add files using upload-large-folder tool 5 days ago
config.pkl
Detected Pickle imports (69)
- "cosmos_policy._src.predict2.networks.minimal_v1_lvg_dit.MinimalV1LVGDiT",
- "cosmos_policy._src.predict2.models.text2world_model.EMAConfig",
- "cosmos_policy._src.imaginaire.config.TrainerConfig",
- "cosmos_policy.models.policy_video2world_model.CosmosPolicyVideo2WorldModel",
- "cosmos_policy._src.predict2.utils.optim_instantiate.get_base_optimizer",
- "omegaconf.nodes.IntegerNode",
- "omegaconf.nodes.StringNode",
- "cosmos_policy.models.policy_video2world_model.CosmosPolicyVideo2WorldConfig",
- "cosmos_policy._src.predict2.checkpointer.dcp.DistributedCheckpointer",
- "typing.Dict",
- "torch.float32",
- "cosmos_policy.config.config_v2.ConfigV2",
- "cosmos_policy._src.predict2.callbacks.compile_tokenizer.CompileTokenizer",
- "megatron.core.parallel_state.get_data_parallel_world_size",
- "omegaconf.dictconfig.DictConfig",
- "cosmos_policy._src.predict2.datasets.cached_replay_dataloader.get_cached_replay_dataloader",
- "cosmos_policy._src.imaginaire.config.JITConfig",
- "cosmos_policy._src.imaginaire.config.Profiling",
- "cosmos_policy.tokenizers.wan2pt1.Wan2pt1VAEInterface",
- "cosmos_policy._src.predict2.datasets.data_sources.mock_data.get_video_dataset",
- "typing.List",
- "cosmos_policy._src.imaginaire.config.ObjectStoreConfig",
- "cosmos_policy.trainer.CosmosPolicyTrainer",
- "omegaconf.base.Metadata",
- "omegaconf.nodes.AnyNode",
- "omegaconf.nodes.FloatNode",
- "megatron.core.model_parallel_config.ModelParallelConfig",
- "builtins.dict",
- "cosmos_policy.modules.hybrid_edm_sde.HybridEDMSDE",
- "cosmos_policy._src.predict2.callbacks.grad_clip.GradClip",
- "builtins.bool",
- "builtins.float",
- "typing.Any",
- "cosmos_policy._src.predict2.callbacks.heart_beat.HeartBeat",
- "cosmos_policy._src.predict2.callbacks.iter_speed.IterSpeed",
- "omegaconf.nodes.BooleanNode",
- "cosmos_policy._src.imaginaire.config.StragglerDetectionConfig",
- "cosmos_policy._src.predict2.datasets.cached_replay_dataloader.duplicate_batches_random",
- "cosmos_policy.datasets.knot_tying_dataset.KnotTyingDataset",
- "torch._utils.torch.preserve_format",
- "cosmos_policy._src.predict2.conditioner.BooleanFlag",
- "omegaconf.listconfig.ListConfig",
- "cosmos_policy._src.predict2.callbacks.device_monitor.DeviceMonitor",
- "cosmos_policy._src.predict2.conditioner.TextAttr",
- "builtins.list",
- "torch.utils.data.dataloader.DataLoader",
- "cosmos_policy._src.predict2.datasets.joint_dataloader.IterativeJointDataLoader",
- "cosmos_policy._src.predict2.networks.minimal_v4_dit.SACConfig",
- "cosmos_policy.config.conditioner.video2world_conditioner.Video2WorldConditioner",
- "builtins.str",
- "functools.partial",
- "cosmos_policy._src.predict2.datasets.data_sources.mock_data.get_image_dataset",
- "builtins.int",
- "cosmos_policy._src.imaginaire.callbacks.manual_gc.ManualGarbageCollection",
- "omegaconf.base.ContainerMetadata",
- "cosmos_policy._src.imaginaire.utils.callback.LowPrecisionCallback",
- "cosmos_policy._src.imaginaire.config.CheckpointConfig",
- "collections.defaultdict",
- "cosmos_policy._src.predict2.callbacks.dataloading_monitor.DetailedDataLoadingSpeedMonitor",
- "cosmos_policy._src.imaginaire.functional.lr_scheduler.LambdaLinearScheduler",
- "cosmos_policy._src.imaginaire.config.CuDNNConfig",
- "cosmos_policy._src.predict2.conditioner.ReMapkey",
- "cosmos_policy._src.predict2.text_encoders.text_encoder.TextEncoderConfig",
- "cosmos_policy._src.predict2.datasets.cached_replay_dataloader.duplicate_batches",
- "cosmos_policy.config.callbacks.WandbCallback",
- "cosmos_policy._src.imaginaire.config.DDPConfig",
- "cosmos_policy._src.imaginaire.config.JobConfig",
- "megatron.core.parallel_state.get_data_parallel_rank",
- "torch.utils.data.distributed.DistributedSampler"
How to fix it?
86 kB
xet

Add files using upload-large-folder tool 5 days ago
config.yaml

17.6 kB
Add files using upload-large-folder tool 5 days ago
job_env.yaml

203 Bytes
Add files using upload-large-folder tool 5 days ago
launch_info.yaml

302 Bytes
Add files using upload-large-folder tool 5 days ago
wandb_id.txt

9 Bytes
Add files using upload-large-folder tool 5 days ago

Detected Pickle imports (69)