├── .gitignore ├── ATTRIBUTIONS.md ├── CONTRIBUTING.md ├── Dockerfile ├── INSTALL.md ├── LICENSE ├── README.md ├── asset ├── examples │ ├── hdri_examples │ │ ├── pink_sunrise_2k.hdr │ │ ├── rosendal_plains_1_2k.hdr │ │ ├── street_lamp_2k.hdr │ │ └── sunny_vondelpark_2k.hdr │ ├── image_examples │ │ ├── image_1.jpg │ │ ├── image_10.jpg │ │ ├── image_11.jpg │ │ ├── image_12.jpg │ │ ├── image_13.jpg │ │ ├── image_14.jpg │ │ ├── image_15.jpg │ │ ├── image_16.jpg │ │ ├── image_2.jpg │ │ ├── image_3.jpg │ │ ├── image_4.jpg │ │ ├── image_5.jpg │ │ ├── image_6.jpg │ │ ├── image_7.jpg │ │ ├── image_8.jpg │ │ └── image_9.jpg │ └── video_examples │ │ ├── .DS_Store │ │ ├── video1.mp4 │ │ ├── video2.mp4 │ │ ├── video3.mp4 │ │ └── video4.mp4 └── teaser.gif ├── checkpoints └── README.md ├── cosmos-predict1.yaml ├── cosmos_predict1 ├── __init__.py ├── autoregressive │ ├── __init__.py │ ├── callbacks │ │ └── video_sampling_teacher_forcing.py │ ├── configs │ │ ├── __init__.py │ │ ├── base │ │ │ ├── __init__.py │ │ │ ├── callbacks.py │ │ │ ├── dataloader.py │ │ │ ├── dataset.py │ │ │ ├── model.py │ │ │ ├── model_config.py │ │ │ ├── model_parallel.py │ │ │ ├── optim.py │ │ │ └── tokenizer.py │ │ ├── config.py │ │ ├── experiment │ │ │ └── video2video │ │ │ │ ├── __init__.py │ │ │ │ └── basic.py │ │ ├── inference │ │ │ └── inference_config.py │ │ └── registry.py │ ├── datasets │ │ ├── dataset_utils.py │ │ └── video_dataset.py │ ├── diffusion_decoder │ │ ├── __init__.py │ │ ├── config │ │ │ ├── base │ │ │ │ └── conditioner.py │ │ │ ├── config_latent_diffusion_decoder.py │ │ │ ├── inference │ │ │ │ └── cosmos_diffusiondecoder_7b.py │ │ │ └── registry.py │ │ ├── inference.py │ │ ├── model.py │ │ ├── network.py │ │ └── utils.py │ ├── inference │ │ ├── __init__.py │ │ ├── base.py │ │ ├── video2world.py │ │ └── world_generation_pipeline.py │ ├── model.py │ ├── modules │ │ ├── __init__.py │ │ ├── attention.py │ │ ├── embedding.py │ │ ├── linear.py │ │ ├── mlp.py │ │ ├── mm_projector.py │ │ └── normalization.py │ ├── networks │ │ ├── transformer.py │ │ └── vit.py │ ├── tokenizer │ │ ├── __init__.py │ │ ├── discrete_video.py │ │ ├── image_text_tokenizer.py │ │ ├── modules.py │ │ ├── networks.py │ │ ├── patching.py │ │ ├── quantizers.py │ │ ├── text_tokenizer.py │ │ ├── tokenizer.py │ │ └── utils.py │ ├── train.py │ ├── trainer.py │ ├── training │ │ ├── model.py │ │ ├── modules │ │ │ └── attention.py │ │ └── networks │ │ │ └── transformer.py │ └── utils │ │ ├── __init__.py │ │ ├── checkpoint.py │ │ ├── inference.py │ │ ├── misc.py │ │ ├── parallel.py │ │ └── sampling.py ├── auxiliary │ ├── guardrail │ │ ├── __init__.py │ │ ├── aegis │ │ │ ├── __init__.py │ │ │ ├── aegis.py │ │ │ └── categories.py │ │ ├── blocklist │ │ │ ├── __init__.py │ │ │ ├── blocklist.py │ │ │ └── utils.py │ │ ├── common │ │ │ ├── __init__.py │ │ │ ├── core.py │ │ │ ├── io_utils.py │ │ │ └── presets.py │ │ ├── face_blur_filter │ │ │ ├── __init__.py │ │ │ ├── blur_utils.py │ │ │ ├── face_blur_filter.py │ │ │ └── retinaface_utils.py │ │ ├── llamaGuard3 │ │ │ ├── __init__.py │ │ │ ├── categories.py │ │ │ └── llamaGuard3.py │ │ └── video_content_safety_filter │ │ │ ├── __init__.py │ │ │ ├── model.py │ │ │ ├── video_content_safety_filter.py │ │ │ └── vision_encoder.py │ └── t5_text_encoder.py ├── callbacks │ ├── every_n.py │ └── grad_clip.py ├── checkpointer │ ├── __init__.py │ ├── base.py │ ├── ddp.py │ ├── peft_checkpointer.py │ ├── safe_broadcast.py │ └── tp.py ├── diffusion │ ├── checkpointers │ │ └── ema_fsdp_checkpointer.py │ ├── conditioner.py │ ├── config │ │ ├── __init__.py │ │ ├── base │ │ │ ├── __init__.py │ │ │ ├── conditioner.py │ │ │ ├── model.py │ │ │ ├── net.py │ │ │ └── tokenizer.py │ │ ├── config.py │ │ ├── diffusion_renderer_config.py │ │ ├── inference │ │ │ ├── __init__.py │ │ │ ├── cosmos-1-diffusion-renderer.py │ │ │ ├── cosmos-1-diffusion-text2world-multiview.py │ │ │ ├── cosmos-1-diffusion-text2world.py │ │ │ ├── cosmos-1-diffusion-video2world-action.py │ │ │ ├── cosmos-1-diffusion-video2world-multiview.py │ │ │ ├── cosmos-1-diffusion-video2world-view_extend-multiview.py │ │ │ ├── cosmos-1-diffusion-video2world.py │ │ │ └── cosmos-1-diffusion-world-interpolator.py │ │ └── registry.py │ ├── functional │ │ ├── batch_ops.py │ │ ├── multi_step.py │ │ └── runge_kutta.py │ ├── inference │ │ ├── diffusion_renderer_pipeline.py │ │ ├── diffusion_renderer_utils │ │ │ ├── __init__.py │ │ │ ├── dataloader_utils.py │ │ │ ├── dataset_inference.py │ │ │ ├── inference_utils.py │ │ │ ├── rendering_utils.py │ │ │ ├── transform_utils.py │ │ │ └── utils_env_proj.py │ │ ├── inference_forward_renderer.py │ │ ├── inference_inverse_renderer.py │ │ ├── inference_utils.py │ │ ├── text2world.py │ │ ├── text2world_multiview.py │ │ ├── video2world.py │ │ ├── video2world_action.py │ │ ├── video2world_multiview.py │ │ ├── video2world_view_extend_multiview.py │ │ ├── world_generation_pipeline.py │ │ └── world_interpolator.py │ ├── model │ │ ├── bu_model_world_interpolator.py │ │ ├── model_diffusion_renderer.py │ │ ├── model_t2w.py │ │ ├── model_t2w_multiview.py │ │ ├── model_v2w.py │ │ ├── model_v2w_action.py │ │ ├── model_v2w_multiview.py │ │ ├── model_view_extend_multiview.py │ │ └── model_world_interpolator.py │ ├── module │ │ ├── __init__.py │ │ ├── attention.py │ │ ├── blocks.py │ │ ├── parallel.py │ │ ├── position_embedding.py │ │ ├── pretrained_vae.py │ │ └── timm.py │ ├── modules │ │ ├── denoiser_scaling.py │ │ └── res_sampler.py │ ├── networks │ │ ├── __init__.py │ │ ├── general_dit.py │ │ ├── general_dit_diffusion_renderer.py │ │ ├── general_dit_multiview.py │ │ ├── general_dit_video_conditioned.py │ │ ├── general_dit_video_conditioned_multiview.py │ │ └── general_dit_view_extend_multiview.py │ ├── prompt_upsampler │ │ ├── inference.py │ │ ├── text2world_prompt_upsampler_inference.py │ │ └── video2world_prompt_upsampler_inference.py │ ├── training │ │ ├── callbacks │ │ │ ├── every_n.py │ │ │ ├── grad_clip.py │ │ │ ├── iter_speed.py │ │ │ └── low_precision.py │ │ ├── conditioner.py │ │ ├── config │ │ │ ├── base │ │ │ │ ├── ema.py │ │ │ │ ├── model.py │ │ │ │ ├── optim.py │ │ │ │ └── vae.py │ │ │ ├── config.py │ │ │ ├── config_multiview.py │ │ │ ├── text2world │ │ │ │ ├── experiment.py │ │ │ │ └── registry.py │ │ │ ├── text2world_multiview │ │ │ │ ├── experiment.py │ │ │ │ └── registry.py │ │ │ ├── text2world_singletomultiview │ │ │ │ ├── experiment.py │ │ │ │ └── registry.py │ │ │ ├── video2world │ │ │ │ ├── experiment.py │ │ │ │ └── registry.py │ │ │ ├── video2world_action │ │ │ │ ├── experiment.py │ │ │ │ └── registry.py │ │ │ ├── video2world_instruction │ │ │ │ ├── experiment.py │ │ │ │ └── registry.py │ │ │ ├── video2world_multiview │ │ │ │ ├── experiment.py │ │ │ │ └── registry.py │ │ │ ├── video2world_singletomultiview │ │ │ │ ├── experiment.py │ │ │ │ └── registry.py │ │ │ └── world_interpolator │ │ │ │ ├── experiment.py │ │ │ │ └── registry.py │ │ ├── context_parallel.py │ │ ├── datasets │ │ │ ├── data_sources │ │ │ │ └── item_dataset.py │ │ │ ├── dataset_3D.py │ │ │ ├── dataset_3D_binary.py │ │ │ ├── dataset_multiview.py │ │ │ └── dataset_video.py │ │ ├── functional │ │ │ ├── loss.py │ │ │ └── lr_scheduler.py │ │ ├── models │ │ │ ├── extend_model.py │ │ │ ├── extend_model_multiview.py │ │ │ ├── interpolator.py │ │ │ ├── model.py │ │ │ ├── model_image.py │ │ │ ├── model_multiview.py │ │ │ ├── model_peft.py │ │ │ └── view_extend_model_multiview.py │ │ ├── module │ │ │ ├── blocks.py │ │ │ ├── position_embedding.py │ │ │ ├── pretrained_vae.py │ │ │ └── pretrained_vae_base.py │ │ ├── modules │ │ │ └── edm_sde.py │ │ ├── networks │ │ │ ├── general_dit.py │ │ │ ├── general_dit_action.py │ │ │ ├── general_dit_lvg.py │ │ │ ├── general_dit_lvg_multiview.py │ │ │ ├── general_dit_multiview.py │ │ │ └── general_dit_view_extend_multiview.py │ │ ├── tensor_parallel.py │ │ ├── train.py │ │ ├── trainer.py │ │ └── utils │ │ │ ├── checkpointer.py │ │ │ ├── fsdp_helper.py │ │ │ ├── inference_long_video.py │ │ │ ├── layer_control │ │ │ └── peft_control_config_parser.py │ │ │ ├── optim_instantiate.py │ │ │ └── peft │ │ │ ├── lora_attn.py │ │ │ ├── lora_attn_test.py │ │ │ ├── lora_config.py │ │ │ ├── lora_net.py │ │ │ └── peft.py │ ├── types.py │ └── utils │ │ ├── customization │ │ └── customization_manager.py │ │ └── dataset_utils.py ├── tokenizer │ ├── __init__.py │ ├── inference │ │ ├── __init__.py │ │ ├── image_cli.py │ │ ├── image_lib.py │ │ ├── utils.py │ │ ├── video_cli.py │ │ └── video_lib.py │ ├── modules │ │ ├── __init__.py │ │ ├── distributions.py │ │ ├── layers2d.py │ │ ├── layers2d_test.py │ │ ├── layers3d.py │ │ ├── layers3d_test.py │ │ ├── patching.py │ │ ├── quantizers.py │ │ └── utils.py │ ├── networks │ │ ├── __init__.py │ │ ├── configs.py │ │ ├── continuous_image.py │ │ ├── continuous_video.py │ │ ├── discrete_image.py │ │ └── discrete_video.py │ ├── notebook │ │ ├── Image_Tokenization.ipynb │ │ └── Video_Tokenization.ipynb │ ├── test_data │ │ ├── image.png │ │ └── video.mp4 │ └── training │ │ ├── __init__.py │ │ ├── callbacks.py │ │ ├── checkpointer.py │ │ ├── configs │ │ ├── __init__.py │ │ ├── base │ │ │ ├── __init__.py │ │ │ ├── callback.py │ │ │ ├── checkpoint.py │ │ │ ├── data.py │ │ │ ├── loss.py │ │ │ ├── metric.py │ │ │ ├── mock_data.py │ │ │ ├── model.py │ │ │ ├── net.py │ │ │ └── optim.py │ │ ├── config.py │ │ ├── experiments │ │ │ ├── __init__.py │ │ │ ├── basic.py │ │ │ ├── cosmos_tokenize1.py │ │ │ └── utils.py │ │ └── registry.py │ │ ├── datasets │ │ ├── __init__.py │ │ ├── augmentation_provider.py │ │ ├── augmentors.py │ │ ├── dataset_provider.py │ │ ├── mock_dataset.py │ │ ├── utils.py │ │ └── video_dataset.py │ │ ├── jit_cli.py │ │ ├── losses │ │ ├── __init__.py │ │ ├── continuous.py │ │ └── lpips.py │ │ ├── metrics.py │ │ ├── model.py │ │ ├── train.py │ │ └── trainer.py └── utils │ ├── __init__.py │ ├── base_world_generation_pipeline.py │ ├── callback.py │ ├── callbacks │ └── grad_clip.py │ ├── checkpointer.py │ ├── config.py │ ├── config_helper.py │ ├── ddp_checkpointer.py │ ├── device.py │ ├── distributed.py │ ├── easy_io │ ├── __init__.py │ ├── backends │ │ ├── __init__.py │ │ ├── base_backend.py │ │ ├── http_backend.py │ │ ├── local_backend.py │ │ └── registry_utils.py │ ├── easy_io.py │ ├── file_client.py │ └── handlers │ │ ├── __init__.py │ │ ├── base.py │ │ ├── csv_handler.py │ │ ├── gzip_handler.py │ │ ├── imageio_video_handler.py │ │ ├── json_handler.py │ │ ├── jsonl_handler.py │ │ ├── np_handler.py │ │ ├── pandas_handler.py │ │ ├── pickle_handler.py │ │ ├── pil_handler.py │ │ ├── registry_utils.py │ │ ├── tarfile_handler.py │ │ ├── torch_handler.py │ │ ├── torchjit_handler.py │ │ ├── txt_handler.py │ │ └── yaml_handler.py │ ├── ema.py │ ├── env_parsers │ ├── cred_env_parser.py │ └── env_parser.py │ ├── fsdp_checkpointer.py │ ├── fsdp_optim_fix.py │ ├── fused_adam.py │ ├── io.py │ ├── lazy_config │ ├── __init__.py │ ├── file_io.py │ ├── instantiate.py │ ├── lazy.py │ ├── omegaconf_patch.py │ └── registry.py │ ├── log.py │ ├── misc.py │ ├── model.py │ ├── parallel_state_helper.py │ ├── scheduler.py │ ├── trainer.py │ ├── validator.py │ └── visualize │ └── video.py ├── datasets └── README.md ├── pyproject.toml ├── requirements.txt └── scripts ├── check_video_links.py ├── dataproc_extract_frames_from_video.py ├── download_autoregressive_checkpoints.py ├── download_diffusion_checkpoints.py ├── download_diffusion_example_data.py ├── download_diffusion_renderer_checkpoints.py ├── download_guardrail_checkpoints.py ├── download_tokenizer_checkpoints.py ├── download_tokenizer_example_data.py ├── format.sh ├── get_t5_embeddings.py ├── get_t5_embeddings_from_bridge.py ├── get_t5_embeddings_from_cosmos_nemo_assets.py ├── get_t5_embeddings_from_waymo.py ├── ip_header.py ├── merge_autoregressive_tp_checkpoints.py ├── shard_autoregressive_base_checkpoints.py └── test_environment.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/.gitignore -------------------------------------------------------------------------------- /ATTRIBUTIONS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/ATTRIBUTIONS.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/Dockerfile -------------------------------------------------------------------------------- /INSTALL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/INSTALL.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/README.md -------------------------------------------------------------------------------- /asset/examples/hdri_examples/pink_sunrise_2k.hdr: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/asset/examples/hdri_examples/pink_sunrise_2k.hdr -------------------------------------------------------------------------------- /asset/examples/hdri_examples/rosendal_plains_1_2k.hdr: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/asset/examples/hdri_examples/rosendal_plains_1_2k.hdr -------------------------------------------------------------------------------- /asset/examples/hdri_examples/street_lamp_2k.hdr: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/asset/examples/hdri_examples/street_lamp_2k.hdr -------------------------------------------------------------------------------- /asset/examples/hdri_examples/sunny_vondelpark_2k.hdr: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/asset/examples/hdri_examples/sunny_vondelpark_2k.hdr -------------------------------------------------------------------------------- /asset/examples/image_examples/image_1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/asset/examples/image_examples/image_1.jpg -------------------------------------------------------------------------------- /asset/examples/image_examples/image_10.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/asset/examples/image_examples/image_10.jpg -------------------------------------------------------------------------------- /asset/examples/image_examples/image_11.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/asset/examples/image_examples/image_11.jpg -------------------------------------------------------------------------------- /asset/examples/image_examples/image_12.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/asset/examples/image_examples/image_12.jpg -------------------------------------------------------------------------------- /asset/examples/image_examples/image_13.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/asset/examples/image_examples/image_13.jpg -------------------------------------------------------------------------------- /asset/examples/image_examples/image_14.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/asset/examples/image_examples/image_14.jpg -------------------------------------------------------------------------------- /asset/examples/image_examples/image_15.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/asset/examples/image_examples/image_15.jpg -------------------------------------------------------------------------------- /asset/examples/image_examples/image_16.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/asset/examples/image_examples/image_16.jpg -------------------------------------------------------------------------------- /asset/examples/image_examples/image_2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/asset/examples/image_examples/image_2.jpg -------------------------------------------------------------------------------- /asset/examples/image_examples/image_3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/asset/examples/image_examples/image_3.jpg -------------------------------------------------------------------------------- /asset/examples/image_examples/image_4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/asset/examples/image_examples/image_4.jpg -------------------------------------------------------------------------------- /asset/examples/image_examples/image_5.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/asset/examples/image_examples/image_5.jpg -------------------------------------------------------------------------------- /asset/examples/image_examples/image_6.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/asset/examples/image_examples/image_6.jpg -------------------------------------------------------------------------------- /asset/examples/image_examples/image_7.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/asset/examples/image_examples/image_7.jpg -------------------------------------------------------------------------------- /asset/examples/image_examples/image_8.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/asset/examples/image_examples/image_8.jpg -------------------------------------------------------------------------------- /asset/examples/image_examples/image_9.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/asset/examples/image_examples/image_9.jpg -------------------------------------------------------------------------------- /asset/examples/video_examples/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/asset/examples/video_examples/.DS_Store -------------------------------------------------------------------------------- /asset/examples/video_examples/video1.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/asset/examples/video_examples/video1.mp4 -------------------------------------------------------------------------------- /asset/examples/video_examples/video2.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/asset/examples/video_examples/video2.mp4 -------------------------------------------------------------------------------- /asset/examples/video_examples/video3.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/asset/examples/video_examples/video3.mp4 -------------------------------------------------------------------------------- /asset/examples/video_examples/video4.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/asset/examples/video_examples/video4.mp4 -------------------------------------------------------------------------------- /asset/teaser.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/asset/teaser.gif -------------------------------------------------------------------------------- /checkpoints/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/checkpoints/README.md -------------------------------------------------------------------------------- /cosmos-predict1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos-predict1.yaml -------------------------------------------------------------------------------- /cosmos_predict1/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/callbacks/video_sampling_teacher_forcing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/callbacks/video_sampling_teacher_forcing.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/configs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/configs/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/configs/base/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/configs/base/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/configs/base/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/configs/base/callbacks.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/configs/base/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/configs/base/dataloader.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/configs/base/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/configs/base/dataset.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/configs/base/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/configs/base/model.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/configs/base/model_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/configs/base/model_config.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/configs/base/model_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/configs/base/model_parallel.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/configs/base/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/configs/base/optim.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/configs/base/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/configs/base/tokenizer.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/configs/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/configs/config.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/configs/experiment/video2video/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/configs/experiment/video2video/basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/configs/experiment/video2video/basic.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/configs/inference/inference_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/configs/inference/inference_config.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/configs/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/configs/registry.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/datasets/dataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/datasets/dataset_utils.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/datasets/video_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/datasets/video_dataset.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/diffusion_decoder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/diffusion_decoder/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/diffusion_decoder/config/base/conditioner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/diffusion_decoder/config/base/conditioner.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/diffusion_decoder/config/config_latent_diffusion_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/diffusion_decoder/config/config_latent_diffusion_decoder.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/diffusion_decoder/config/inference/cosmos_diffusiondecoder_7b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/diffusion_decoder/config/inference/cosmos_diffusiondecoder_7b.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/diffusion_decoder/config/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/diffusion_decoder/config/registry.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/diffusion_decoder/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/diffusion_decoder/inference.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/diffusion_decoder/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/diffusion_decoder/model.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/diffusion_decoder/network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/diffusion_decoder/network.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/diffusion_decoder/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/diffusion_decoder/utils.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/inference/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/inference/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/inference/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/inference/base.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/inference/video2world.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/inference/video2world.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/inference/world_generation_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/inference/world_generation_pipeline.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/model.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/modules/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/modules/attention.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/modules/embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/modules/embedding.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/modules/linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/modules/linear.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/modules/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/modules/mlp.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/modules/mm_projector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/modules/mm_projector.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/modules/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/modules/normalization.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/networks/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/networks/transformer.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/networks/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/networks/vit.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/tokenizer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/tokenizer/discrete_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/tokenizer/discrete_video.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/tokenizer/image_text_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/tokenizer/image_text_tokenizer.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/tokenizer/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/tokenizer/modules.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/tokenizer/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/tokenizer/networks.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/tokenizer/patching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/tokenizer/patching.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/tokenizer/quantizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/tokenizer/quantizers.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/tokenizer/text_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/tokenizer/text_tokenizer.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/tokenizer/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/tokenizer/tokenizer.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/tokenizer/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/tokenizer/utils.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/train.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/trainer.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/training/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/training/model.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/training/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/training/modules/attention.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/training/networks/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/training/networks/transformer.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/utils/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/utils/checkpoint.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/utils/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/utils/inference.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/utils/misc.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/utils/parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/utils/parallel.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/utils/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/autoregressive/utils/sampling.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/auxiliary/guardrail/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/aegis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/auxiliary/guardrail/aegis/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/aegis/aegis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/auxiliary/guardrail/aegis/aegis.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/aegis/categories.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/auxiliary/guardrail/aegis/categories.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/blocklist/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/auxiliary/guardrail/blocklist/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/blocklist/blocklist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/auxiliary/guardrail/blocklist/blocklist.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/blocklist/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/auxiliary/guardrail/blocklist/utils.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/common/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/common/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/auxiliary/guardrail/common/core.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/common/io_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/auxiliary/guardrail/common/io_utils.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/common/presets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/auxiliary/guardrail/common/presets.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/face_blur_filter/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/auxiliary/guardrail/face_blur_filter/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/face_blur_filter/blur_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/auxiliary/guardrail/face_blur_filter/blur_utils.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/face_blur_filter/face_blur_filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/auxiliary/guardrail/face_blur_filter/face_blur_filter.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/face_blur_filter/retinaface_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/auxiliary/guardrail/face_blur_filter/retinaface_utils.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/llamaGuard3/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/auxiliary/guardrail/llamaGuard3/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/llamaGuard3/categories.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/auxiliary/guardrail/llamaGuard3/categories.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/llamaGuard3/llamaGuard3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/auxiliary/guardrail/llamaGuard3/llamaGuard3.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/video_content_safety_filter/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/auxiliary/guardrail/video_content_safety_filter/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/video_content_safety_filter/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/auxiliary/guardrail/video_content_safety_filter/model.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/video_content_safety_filter/video_content_safety_filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/auxiliary/guardrail/video_content_safety_filter/video_content_safety_filter.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/video_content_safety_filter/vision_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/auxiliary/guardrail/video_content_safety_filter/vision_encoder.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/t5_text_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/auxiliary/t5_text_encoder.py -------------------------------------------------------------------------------- /cosmos_predict1/callbacks/every_n.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/callbacks/every_n.py -------------------------------------------------------------------------------- /cosmos_predict1/callbacks/grad_clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/callbacks/grad_clip.py -------------------------------------------------------------------------------- /cosmos_predict1/checkpointer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/checkpointer/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/checkpointer/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/checkpointer/base.py -------------------------------------------------------------------------------- /cosmos_predict1/checkpointer/ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/checkpointer/ddp.py -------------------------------------------------------------------------------- /cosmos_predict1/checkpointer/peft_checkpointer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/checkpointer/peft_checkpointer.py -------------------------------------------------------------------------------- /cosmos_predict1/checkpointer/safe_broadcast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/checkpointer/safe_broadcast.py -------------------------------------------------------------------------------- /cosmos_predict1/checkpointer/tp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/checkpointer/tp.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/checkpointers/ema_fsdp_checkpointer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/checkpointers/ema_fsdp_checkpointer.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/conditioner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/conditioner.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/config/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/config/base/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/config/base/conditioner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/config/base/conditioner.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/config/base/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/config/base/model.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/config/base/net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/config/base/net.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/config/base/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/config/base/tokenizer.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/config/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/config/config.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/config/diffusion_renderer_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/config/diffusion_renderer_config.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/config/inference/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/config/inference/cosmos-1-diffusion-renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/config/inference/cosmos-1-diffusion-renderer.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/config/inference/cosmos-1-diffusion-text2world-multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/config/inference/cosmos-1-diffusion-text2world-multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/config/inference/cosmos-1-diffusion-text2world.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/config/inference/cosmos-1-diffusion-text2world.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/config/inference/cosmos-1-diffusion-video2world-action.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/config/inference/cosmos-1-diffusion-video2world-action.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/config/inference/cosmos-1-diffusion-video2world-multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/config/inference/cosmos-1-diffusion-video2world-multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/config/inference/cosmos-1-diffusion-video2world-view_extend-multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/config/inference/cosmos-1-diffusion-video2world-view_extend-multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/config/inference/cosmos-1-diffusion-video2world.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/config/inference/cosmos-1-diffusion-video2world.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/config/inference/cosmos-1-diffusion-world-interpolator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/config/inference/cosmos-1-diffusion-world-interpolator.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/config/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/config/registry.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/functional/batch_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/functional/batch_ops.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/functional/multi_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/functional/multi_step.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/functional/runge_kutta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/functional/runge_kutta.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/diffusion_renderer_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/inference/diffusion_renderer_pipeline.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/diffusion_renderer_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/inference/diffusion_renderer_utils/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/diffusion_renderer_utils/dataloader_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/inference/diffusion_renderer_utils/dataloader_utils.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/diffusion_renderer_utils/dataset_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/inference/diffusion_renderer_utils/dataset_inference.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/diffusion_renderer_utils/inference_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/inference/diffusion_renderer_utils/inference_utils.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/diffusion_renderer_utils/rendering_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/inference/diffusion_renderer_utils/rendering_utils.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/diffusion_renderer_utils/transform_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/inference/diffusion_renderer_utils/transform_utils.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/diffusion_renderer_utils/utils_env_proj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/inference/diffusion_renderer_utils/utils_env_proj.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/inference_forward_renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/inference/inference_forward_renderer.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/inference_inverse_renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/inference/inference_inverse_renderer.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/inference_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/inference/inference_utils.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/text2world.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/inference/text2world.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/text2world_multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/inference/text2world_multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/video2world.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/inference/video2world.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/video2world_action.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/inference/video2world_action.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/video2world_multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/inference/video2world_multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/video2world_view_extend_multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/inference/video2world_view_extend_multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/world_generation_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/inference/world_generation_pipeline.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/world_interpolator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/inference/world_interpolator.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/model/bu_model_world_interpolator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/model/bu_model_world_interpolator.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/model/model_diffusion_renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/model/model_diffusion_renderer.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/model/model_t2w.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/model/model_t2w.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/model/model_t2w_multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/model/model_t2w_multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/model/model_v2w.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/model/model_v2w.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/model/model_v2w_action.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/model/model_v2w_action.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/model/model_v2w_multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/model/model_v2w_multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/model/model_view_extend_multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/model/model_view_extend_multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/model/model_world_interpolator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/model/model_world_interpolator.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/module/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/module/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/module/attention.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/module/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/module/blocks.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/module/parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/module/parallel.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/module/position_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/module/position_embedding.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/module/pretrained_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/module/pretrained_vae.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/module/timm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/module/timm.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/modules/denoiser_scaling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/modules/denoiser_scaling.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/modules/res_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/modules/res_sampler.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/networks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/networks/general_dit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/networks/general_dit.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/networks/general_dit_diffusion_renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/networks/general_dit_diffusion_renderer.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/networks/general_dit_multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/networks/general_dit_multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/networks/general_dit_video_conditioned.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/networks/general_dit_video_conditioned.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/networks/general_dit_video_conditioned_multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/networks/general_dit_video_conditioned_multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/networks/general_dit_view_extend_multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/networks/general_dit_view_extend_multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/prompt_upsampler/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/prompt_upsampler/inference.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/prompt_upsampler/text2world_prompt_upsampler_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/prompt_upsampler/text2world_prompt_upsampler_inference.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/prompt_upsampler/video2world_prompt_upsampler_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/prompt_upsampler/video2world_prompt_upsampler_inference.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/callbacks/every_n.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/callbacks/every_n.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/callbacks/grad_clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/callbacks/grad_clip.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/callbacks/iter_speed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/callbacks/iter_speed.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/callbacks/low_precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/callbacks/low_precision.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/conditioner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/conditioner.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/base/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/config/base/ema.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/base/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/config/base/model.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/base/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/config/base/optim.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/base/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/config/base/vae.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/config/config.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/config_multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/config/config_multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/text2world/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/config/text2world/experiment.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/text2world/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/config/text2world/registry.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/text2world_multiview/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/config/text2world_multiview/experiment.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/text2world_multiview/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/config/text2world_multiview/registry.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/text2world_singletomultiview/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/config/text2world_singletomultiview/experiment.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/text2world_singletomultiview/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/config/text2world_singletomultiview/registry.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/video2world/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/config/video2world/experiment.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/video2world/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/config/video2world/registry.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/video2world_action/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/config/video2world_action/experiment.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/video2world_action/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/config/video2world_action/registry.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/video2world_instruction/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/config/video2world_instruction/experiment.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/video2world_instruction/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/config/video2world_instruction/registry.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/video2world_multiview/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/config/video2world_multiview/experiment.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/video2world_multiview/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/config/video2world_multiview/registry.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/video2world_singletomultiview/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/config/video2world_singletomultiview/experiment.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/video2world_singletomultiview/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/config/video2world_singletomultiview/registry.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/world_interpolator/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/config/world_interpolator/experiment.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/world_interpolator/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/config/world_interpolator/registry.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/context_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/context_parallel.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/datasets/data_sources/item_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/datasets/data_sources/item_dataset.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/datasets/dataset_3D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/datasets/dataset_3D.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/datasets/dataset_3D_binary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/datasets/dataset_3D_binary.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/datasets/dataset_multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/datasets/dataset_multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/datasets/dataset_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/datasets/dataset_video.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/functional/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/functional/loss.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/functional/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/functional/lr_scheduler.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/models/extend_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/models/extend_model.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/models/extend_model_multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/models/extend_model_multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/models/interpolator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/models/interpolator.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/models/model.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/models/model_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/models/model_image.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/models/model_multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/models/model_multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/models/model_peft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/models/model_peft.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/models/view_extend_model_multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/models/view_extend_model_multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/module/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/module/blocks.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/module/position_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/module/position_embedding.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/module/pretrained_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/module/pretrained_vae.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/module/pretrained_vae_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/module/pretrained_vae_base.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/modules/edm_sde.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/modules/edm_sde.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/networks/general_dit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/networks/general_dit.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/networks/general_dit_action.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/networks/general_dit_action.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/networks/general_dit_lvg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/networks/general_dit_lvg.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/networks/general_dit_lvg_multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/networks/general_dit_lvg_multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/networks/general_dit_multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/networks/general_dit_multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/networks/general_dit_view_extend_multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/networks/general_dit_view_extend_multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/tensor_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/tensor_parallel.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/train.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/trainer.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/utils/checkpointer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/utils/checkpointer.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/utils/fsdp_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/utils/fsdp_helper.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/utils/inference_long_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/utils/inference_long_video.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/utils/layer_control/peft_control_config_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/utils/layer_control/peft_control_config_parser.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/utils/optim_instantiate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/utils/optim_instantiate.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/utils/peft/lora_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/utils/peft/lora_attn.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/utils/peft/lora_attn_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/utils/peft/lora_attn_test.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/utils/peft/lora_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/utils/peft/lora_config.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/utils/peft/lora_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/utils/peft/lora_net.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/utils/peft/peft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/training/utils/peft/peft.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/types.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/utils/customization/customization_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/utils/customization/customization_manager.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/utils/dataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/diffusion/utils/dataset_utils.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/inference/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/inference/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/inference/image_cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/inference/image_cli.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/inference/image_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/inference/image_lib.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/inference/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/inference/utils.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/inference/video_cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/inference/video_cli.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/inference/video_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/inference/video_lib.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/modules/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/modules/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/modules/distributions.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/modules/layers2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/modules/layers2d.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/modules/layers2d_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/modules/layers2d_test.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/modules/layers3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/modules/layers3d.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/modules/layers3d_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/modules/layers3d_test.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/modules/patching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/modules/patching.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/modules/quantizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/modules/quantizers.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/modules/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/modules/utils.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/networks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/networks/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/networks/configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/networks/configs.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/networks/continuous_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/networks/continuous_image.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/networks/continuous_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/networks/continuous_video.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/networks/discrete_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/networks/discrete_image.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/networks/discrete_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/networks/discrete_video.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/notebook/Image_Tokenization.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/notebook/Image_Tokenization.ipynb -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/notebook/Video_Tokenization.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/notebook/Video_Tokenization.ipynb -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/test_data/image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/test_data/image.png -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/test_data/video.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/test_data/video.mp4 -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/callbacks.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/checkpointer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/checkpointer.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/configs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/configs/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/configs/base/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/configs/base/callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/configs/base/callback.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/configs/base/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/configs/base/checkpoint.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/configs/base/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/configs/base/data.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/configs/base/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/configs/base/loss.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/configs/base/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/configs/base/metric.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/configs/base/mock_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/configs/base/mock_data.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/configs/base/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/configs/base/model.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/configs/base/net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/configs/base/net.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/configs/base/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/configs/base/optim.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/configs/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/configs/config.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/configs/experiments/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/configs/experiments/basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/configs/experiments/basic.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/configs/experiments/cosmos_tokenize1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/configs/experiments/cosmos_tokenize1.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/configs/experiments/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/configs/experiments/utils.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/configs/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/configs/registry.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/datasets/augmentation_provider.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/datasets/augmentation_provider.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/datasets/augmentors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/datasets/augmentors.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/datasets/dataset_provider.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/datasets/dataset_provider.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/datasets/mock_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/datasets/mock_dataset.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/datasets/utils.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/datasets/video_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/datasets/video_dataset.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/jit_cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/jit_cli.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/losses/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/losses/continuous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/losses/continuous.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/losses/lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/losses/lpips.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/metrics.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/model.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/train.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/tokenizer/training/trainer.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/base_world_generation_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/base_world_generation_pipeline.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/callback.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/callbacks/grad_clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/callbacks/grad_clip.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/checkpointer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/checkpointer.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/config.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/config_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/config_helper.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/ddp_checkpointer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/ddp_checkpointer.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/device.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/device.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/distributed.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/easy_io/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/backends/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/easy_io/backends/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/backends/base_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/easy_io/backends/base_backend.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/backends/http_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/easy_io/backends/http_backend.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/backends/local_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/easy_io/backends/local_backend.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/backends/registry_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/easy_io/backends/registry_utils.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/easy_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/easy_io/easy_io.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/file_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/easy_io/file_client.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/handlers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/easy_io/handlers/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/handlers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/easy_io/handlers/base.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/handlers/csv_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/easy_io/handlers/csv_handler.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/handlers/gzip_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/easy_io/handlers/gzip_handler.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/handlers/imageio_video_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/easy_io/handlers/imageio_video_handler.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/handlers/json_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/easy_io/handlers/json_handler.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/handlers/jsonl_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/easy_io/handlers/jsonl_handler.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/handlers/np_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/easy_io/handlers/np_handler.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/handlers/pandas_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/easy_io/handlers/pandas_handler.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/handlers/pickle_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/easy_io/handlers/pickle_handler.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/handlers/pil_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/easy_io/handlers/pil_handler.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/handlers/registry_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/easy_io/handlers/registry_utils.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/handlers/tarfile_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/easy_io/handlers/tarfile_handler.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/handlers/torch_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/easy_io/handlers/torch_handler.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/handlers/torchjit_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/easy_io/handlers/torchjit_handler.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/handlers/txt_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/easy_io/handlers/txt_handler.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/handlers/yaml_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/easy_io/handlers/yaml_handler.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/ema.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/env_parsers/cred_env_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/env_parsers/cred_env_parser.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/env_parsers/env_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/env_parsers/env_parser.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/fsdp_checkpointer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/fsdp_checkpointer.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/fsdp_optim_fix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/fsdp_optim_fix.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/fused_adam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/fused_adam.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/io.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/lazy_config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/lazy_config/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/lazy_config/file_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/lazy_config/file_io.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/lazy_config/instantiate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/lazy_config/instantiate.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/lazy_config/lazy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/lazy_config/lazy.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/lazy_config/omegaconf_patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/lazy_config/omegaconf_patch.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/lazy_config/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/lazy_config/registry.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/log.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/misc.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/model.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/parallel_state_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/parallel_state_helper.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/scheduler.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/trainer.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/validator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/validator.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/visualize/video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/cosmos_predict1/utils/visualize/video.py -------------------------------------------------------------------------------- /datasets/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/datasets/README.md -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/check_video_links.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/scripts/check_video_links.py -------------------------------------------------------------------------------- /scripts/dataproc_extract_frames_from_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/scripts/dataproc_extract_frames_from_video.py -------------------------------------------------------------------------------- /scripts/download_autoregressive_checkpoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/scripts/download_autoregressive_checkpoints.py -------------------------------------------------------------------------------- /scripts/download_diffusion_checkpoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/scripts/download_diffusion_checkpoints.py -------------------------------------------------------------------------------- /scripts/download_diffusion_example_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/scripts/download_diffusion_example_data.py -------------------------------------------------------------------------------- /scripts/download_diffusion_renderer_checkpoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/scripts/download_diffusion_renderer_checkpoints.py -------------------------------------------------------------------------------- /scripts/download_guardrail_checkpoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/scripts/download_guardrail_checkpoints.py -------------------------------------------------------------------------------- /scripts/download_tokenizer_checkpoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/scripts/download_tokenizer_checkpoints.py -------------------------------------------------------------------------------- /scripts/download_tokenizer_example_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/scripts/download_tokenizer_example_data.py -------------------------------------------------------------------------------- /scripts/format.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/scripts/format.sh -------------------------------------------------------------------------------- /scripts/get_t5_embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/scripts/get_t5_embeddings.py -------------------------------------------------------------------------------- /scripts/get_t5_embeddings_from_bridge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/scripts/get_t5_embeddings_from_bridge.py -------------------------------------------------------------------------------- /scripts/get_t5_embeddings_from_cosmos_nemo_assets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/scripts/get_t5_embeddings_from_cosmos_nemo_assets.py -------------------------------------------------------------------------------- /scripts/get_t5_embeddings_from_waymo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/scripts/get_t5_embeddings_from_waymo.py -------------------------------------------------------------------------------- /scripts/ip_header.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/scripts/ip_header.py -------------------------------------------------------------------------------- /scripts/merge_autoregressive_tp_checkpoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/scripts/merge_autoregressive_tp_checkpoints.py -------------------------------------------------------------------------------- /scripts/shard_autoregressive_base_checkpoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/scripts/shard_autoregressive_base_checkpoints.py -------------------------------------------------------------------------------- /scripts/test_environment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/cosmos-transfer1-diffusion-renderer/HEAD/scripts/test_environment.py --------------------------------------------------------------------------------