├── ATTRIBUTIONS.md ├── CONTRIBUTING.md ├── INSTALL.md ├── LICENSE ├── README.md ├── configs ├── accelerate │ ├── accelerate_config.yaml │ └── accelerate_config_single.yaml ├── demo │ ├── lyra_dynamic.yaml │ └── lyra_static.yaml ├── inference │ ├── 3dgs_res_176_320_views_17.yaml │ ├── 3dgs_res_176_320_views_49.yaml │ ├── 3dgs_res_352_640_views_49.yaml │ ├── 3dgs_res_704_1280_views_121.yaml │ ├── 3dgs_res_704_1280_views_121_multi_6.yaml │ ├── 3dgs_res_704_1280_views_121_multi_6_dynamic.yaml │ ├── 3dgs_res_704_1280_views_121_multi_6_dynamic_prune.yaml │ ├── 3dgs_res_704_1280_views_121_multi_6_prune.yaml │ ├── 3dgs_res_704_1280_views_49.yaml │ └── default.yaml └── training │ ├── 3dgs_res_176_320_views_17.yaml │ ├── 3dgs_res_176_320_views_49.yaml │ ├── 3dgs_res_352_640_views_49.yaml │ ├── 3dgs_res_704_1280_views_121.yaml │ ├── 3dgs_res_704_1280_views_121_multi_6.yaml │ ├── 3dgs_res_704_1280_views_121_multi_6_dynamic.yaml │ ├── 3dgs_res_704_1280_views_121_multi_6_dynamic_prune.yaml │ ├── 3dgs_res_704_1280_views_121_multi_6_prune.yaml │ ├── 3dgs_res_704_1280_views_49.yaml │ └── default.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 │ │ ├── inference │ │ │ ├── __init__.py │ │ │ ├── cosmos-1-diffusion-gen3c.py │ │ │ ├── cosmos-1-diffusion-text2world-multiview.py │ │ │ ├── cosmos-1-diffusion-text2world.py │ │ │ ├── cosmos-1-diffusion-video2world-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 │ │ ├── cache_3d.py │ │ ├── camera_utils.py │ │ ├── data_loader_utils.py │ │ ├── forward_warp_utils_pytorch.py │ │ ├── gen3c_dynamic.py │ │ ├── gen3c_dynamic_sdg.py │ │ ├── gen3c_multiview.py │ │ ├── gen3c_persistent.py │ │ ├── gen3c_pipeline.py │ │ ├── gen3c_single_image.py │ │ ├── gen3c_single_image_sdg.py │ │ ├── inference_utils.py │ │ ├── ray_triangle_intersection_warp.py │ │ ├── text2world.py │ │ ├── text2world_multiview.py │ │ ├── video2world.py │ │ ├── video2world_multiview.py │ │ ├── vipe_utils.py │ │ ├── world_generation_pipeline.py │ │ └── world_interpolator.py │ ├── model │ │ ├── bu_model_world_interpolator.py │ │ ├── model_gen3c.py │ │ ├── model_t2w.py │ │ ├── model_t2w_multiview.py │ │ ├── model_v2w.py │ │ ├── model_v2w_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_multiview.py │ │ ├── general_dit_video_conditioned.py │ │ └── general_dit_video_conditioned_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 │ │ │ ├── video2world │ │ │ │ ├── experiment.py │ │ │ │ └── registry.py │ │ │ ├── video2world_action │ │ │ │ ├── experiment.py │ │ │ │ └── registry.py │ │ │ ├── video2world_instruction │ │ │ │ ├── experiment.py │ │ │ │ └── registry.py │ │ │ ├── video2world_multiview │ │ │ │ ├── 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_utils.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 │ │ ├── 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 │ │ ├── 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 ├── 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 │ ├── test_data │ │ └── 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 ├── inference.sh ├── lyra.yaml ├── requirements_gen3c.txt ├── requirements_lyra.txt ├── sample.py ├── scripts ├── bash │ ├── dynamic_sdg.sh │ └── static_sdg.sh ├── download_gen3c_checkpoints.py ├── download_guardrail_checkpoints.py ├── download_lyra_checkpoints.py ├── download_tokenizer_checkpoints.py └── test_environment.py ├── src ├── __init__.py ├── eval │ ├── compute_metrics_datasets.py │ └── metrics.py ├── models │ ├── __init__.py │ ├── data │ │ ├── __init__.py │ │ ├── base.py │ │ ├── datafield.py │ │ ├── provider.py │ │ ├── radym.py │ │ ├── radym_wrapper.py │ │ └── registry.py │ ├── recon │ │ ├── __init__.py │ │ └── model_latent_recon.py │ └── utils │ │ ├── __init__.py │ │ ├── attention.py │ │ ├── cosmos_1_tokenizer.py │ │ ├── data.py │ │ ├── loss.py │ │ ├── mamba2.py │ │ ├── misc.py │ │ ├── model.py │ │ ├── render.py │ │ ├── token_pruning.py │ │ └── train.py ├── rendering │ ├── __init__.py │ ├── gs.py │ ├── gs_deferred.py │ └── gs_deferred_patch.py ├── utils │ ├── copy_to_resume.py │ ├── random_state_utils.py │ └── visu.py └── visu │ ├── dynamic_grid.py │ ├── dynamic_scene_merge.py │ ├── dynamic_scene_video_scene_and_camera.py │ ├── dynamic_scene_video_static_camera.py │ ├── dynamic_scene_video_static_scene.py │ ├── dynamic_visu.sh │ └── static_visu.py ├── train.py └── train.sh /ATTRIBUTIONS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/ATTRIBUTIONS.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /INSTALL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/INSTALL.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/README.md -------------------------------------------------------------------------------- /configs/accelerate/accelerate_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/configs/accelerate/accelerate_config.yaml -------------------------------------------------------------------------------- /configs/accelerate/accelerate_config_single.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/configs/accelerate/accelerate_config_single.yaml -------------------------------------------------------------------------------- /configs/demo/lyra_dynamic.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/configs/demo/lyra_dynamic.yaml -------------------------------------------------------------------------------- /configs/demo/lyra_static.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/configs/demo/lyra_static.yaml -------------------------------------------------------------------------------- /configs/inference/3dgs_res_176_320_views_17.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/configs/inference/3dgs_res_176_320_views_17.yaml -------------------------------------------------------------------------------- /configs/inference/3dgs_res_176_320_views_49.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/configs/inference/3dgs_res_176_320_views_49.yaml -------------------------------------------------------------------------------- /configs/inference/3dgs_res_352_640_views_49.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/configs/inference/3dgs_res_352_640_views_49.yaml -------------------------------------------------------------------------------- /configs/inference/3dgs_res_704_1280_views_121.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/configs/inference/3dgs_res_704_1280_views_121.yaml -------------------------------------------------------------------------------- /configs/inference/3dgs_res_704_1280_views_121_multi_6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/configs/inference/3dgs_res_704_1280_views_121_multi_6.yaml -------------------------------------------------------------------------------- /configs/inference/3dgs_res_704_1280_views_121_multi_6_dynamic.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/configs/inference/3dgs_res_704_1280_views_121_multi_6_dynamic.yaml -------------------------------------------------------------------------------- /configs/inference/3dgs_res_704_1280_views_121_multi_6_dynamic_prune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/configs/inference/3dgs_res_704_1280_views_121_multi_6_dynamic_prune.yaml -------------------------------------------------------------------------------- /configs/inference/3dgs_res_704_1280_views_121_multi_6_prune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/configs/inference/3dgs_res_704_1280_views_121_multi_6_prune.yaml -------------------------------------------------------------------------------- /configs/inference/3dgs_res_704_1280_views_49.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/configs/inference/3dgs_res_704_1280_views_49.yaml -------------------------------------------------------------------------------- /configs/inference/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/configs/inference/default.yaml -------------------------------------------------------------------------------- /configs/training/3dgs_res_176_320_views_17.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/configs/training/3dgs_res_176_320_views_17.yaml -------------------------------------------------------------------------------- /configs/training/3dgs_res_176_320_views_49.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/configs/training/3dgs_res_176_320_views_49.yaml -------------------------------------------------------------------------------- /configs/training/3dgs_res_352_640_views_49.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/configs/training/3dgs_res_352_640_views_49.yaml -------------------------------------------------------------------------------- /configs/training/3dgs_res_704_1280_views_121.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/configs/training/3dgs_res_704_1280_views_121.yaml -------------------------------------------------------------------------------- /configs/training/3dgs_res_704_1280_views_121_multi_6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/configs/training/3dgs_res_704_1280_views_121_multi_6.yaml -------------------------------------------------------------------------------- /configs/training/3dgs_res_704_1280_views_121_multi_6_dynamic.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/configs/training/3dgs_res_704_1280_views_121_multi_6_dynamic.yaml -------------------------------------------------------------------------------- /configs/training/3dgs_res_704_1280_views_121_multi_6_dynamic_prune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/configs/training/3dgs_res_704_1280_views_121_multi_6_dynamic_prune.yaml -------------------------------------------------------------------------------- /configs/training/3dgs_res_704_1280_views_121_multi_6_prune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/configs/training/3dgs_res_704_1280_views_121_multi_6_prune.yaml -------------------------------------------------------------------------------- /configs/training/3dgs_res_704_1280_views_49.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/configs/training/3dgs_res_704_1280_views_49.yaml -------------------------------------------------------------------------------- /configs/training/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/configs/training/default.yaml -------------------------------------------------------------------------------- /cosmos_predict1/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/callbacks/video_sampling_teacher_forcing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/callbacks/video_sampling_teacher_forcing.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/configs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/configs/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/configs/base/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/configs/base/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/configs/base/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/configs/base/callbacks.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/configs/base/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/configs/base/dataloader.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/configs/base/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/configs/base/dataset.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/configs/base/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/configs/base/model.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/configs/base/model_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/configs/base/model_config.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/configs/base/model_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/configs/base/model_parallel.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/configs/base/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/configs/base/optim.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/configs/base/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/configs/base/tokenizer.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/configs/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/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/lyra/HEAD/cosmos_predict1/autoregressive/configs/experiment/video2video/basic.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/configs/inference/inference_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/configs/inference/inference_config.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/configs/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/configs/registry.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/datasets/dataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/datasets/dataset_utils.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/datasets/video_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/datasets/video_dataset.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/diffusion_decoder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/diffusion_decoder/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/diffusion_decoder/config/base/conditioner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/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/lyra/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/lyra/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/lyra/HEAD/cosmos_predict1/autoregressive/diffusion_decoder/config/registry.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/diffusion_decoder/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/diffusion_decoder/inference.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/diffusion_decoder/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/diffusion_decoder/model.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/diffusion_decoder/network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/diffusion_decoder/network.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/diffusion_decoder/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/diffusion_decoder/utils.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/inference/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/inference/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/inference/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/inference/base.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/inference/video2world.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/inference/video2world.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/inference/world_generation_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/inference/world_generation_pipeline.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/model.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/modules/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/modules/attention.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/modules/embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/modules/embedding.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/modules/linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/modules/linear.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/modules/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/modules/mlp.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/modules/mm_projector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/modules/mm_projector.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/modules/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/modules/normalization.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/networks/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/networks/transformer.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/networks/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/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/lyra/HEAD/cosmos_predict1/autoregressive/tokenizer/discrete_video.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/tokenizer/image_text_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/tokenizer/image_text_tokenizer.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/tokenizer/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/tokenizer/modules.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/tokenizer/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/tokenizer/networks.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/tokenizer/patching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/tokenizer/patching.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/tokenizer/quantizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/tokenizer/quantizers.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/tokenizer/text_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/tokenizer/text_tokenizer.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/tokenizer/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/tokenizer/tokenizer.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/tokenizer/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/tokenizer/utils.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/train.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/trainer.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/training/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/training/model.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/training/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/training/modules/attention.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/training/networks/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/training/networks/transformer.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/utils/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/utils/checkpoint.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/utils/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/utils/inference.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/utils/misc.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/utils/parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/utils/parallel.py -------------------------------------------------------------------------------- /cosmos_predict1/autoregressive/utils/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/autoregressive/utils/sampling.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/auxiliary/guardrail/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/aegis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/auxiliary/guardrail/aegis/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/aegis/aegis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/auxiliary/guardrail/aegis/aegis.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/aegis/categories.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/auxiliary/guardrail/aegis/categories.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/blocklist/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/auxiliary/guardrail/blocklist/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/blocklist/blocklist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/auxiliary/guardrail/blocklist/blocklist.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/blocklist/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/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/lyra/HEAD/cosmos_predict1/auxiliary/guardrail/common/core.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/common/io_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/auxiliary/guardrail/common/io_utils.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/common/presets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/auxiliary/guardrail/common/presets.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/face_blur_filter/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/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/lyra/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/lyra/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/lyra/HEAD/cosmos_predict1/auxiliary/guardrail/face_blur_filter/retinaface_utils.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/llamaGuard3/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/auxiliary/guardrail/llamaGuard3/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/llamaGuard3/categories.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/auxiliary/guardrail/llamaGuard3/categories.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/llamaGuard3/llamaGuard3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/auxiliary/guardrail/llamaGuard3/llamaGuard3.py -------------------------------------------------------------------------------- /cosmos_predict1/auxiliary/guardrail/video_content_safety_filter/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/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/lyra/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/lyra/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/lyra/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/lyra/HEAD/cosmos_predict1/auxiliary/t5_text_encoder.py -------------------------------------------------------------------------------- /cosmos_predict1/callbacks/every_n.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/callbacks/every_n.py -------------------------------------------------------------------------------- /cosmos_predict1/callbacks/grad_clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/callbacks/grad_clip.py -------------------------------------------------------------------------------- /cosmos_predict1/checkpointer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/checkpointer/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/checkpointer/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/checkpointer/base.py -------------------------------------------------------------------------------- /cosmos_predict1/checkpointer/ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/checkpointer/ddp.py -------------------------------------------------------------------------------- /cosmos_predict1/checkpointer/peft_checkpointer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/checkpointer/peft_checkpointer.py -------------------------------------------------------------------------------- /cosmos_predict1/checkpointer/safe_broadcast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/checkpointer/safe_broadcast.py -------------------------------------------------------------------------------- /cosmos_predict1/checkpointer/tp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/checkpointer/tp.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/checkpointers/ema_fsdp_checkpointer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/checkpointers/ema_fsdp_checkpointer.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/conditioner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/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/lyra/HEAD/cosmos_predict1/diffusion/config/base/conditioner.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/config/base/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/config/base/model.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/config/base/net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/config/base/net.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/config/base/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/config/base/tokenizer.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/config/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/config/config.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/config/inference/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/config/inference/cosmos-1-diffusion-gen3c.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/config/inference/cosmos-1-diffusion-gen3c.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/config/inference/cosmos-1-diffusion-text2world-multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/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/lyra/HEAD/cosmos_predict1/diffusion/config/inference/cosmos-1-diffusion-text2world.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/config/inference/cosmos-1-diffusion-video2world-multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/config/inference/cosmos-1-diffusion-video2world-multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/config/inference/cosmos-1-diffusion-video2world.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/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/lyra/HEAD/cosmos_predict1/diffusion/config/inference/cosmos-1-diffusion-world-interpolator.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/config/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/config/registry.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/functional/batch_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/functional/batch_ops.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/functional/multi_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/functional/multi_step.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/functional/runge_kutta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/functional/runge_kutta.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/cache_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/inference/cache_3d.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/camera_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/inference/camera_utils.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/data_loader_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/inference/data_loader_utils.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/forward_warp_utils_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/inference/forward_warp_utils_pytorch.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/gen3c_dynamic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/inference/gen3c_dynamic.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/gen3c_dynamic_sdg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/inference/gen3c_dynamic_sdg.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/gen3c_multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/inference/gen3c_multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/gen3c_persistent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/inference/gen3c_persistent.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/gen3c_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/inference/gen3c_pipeline.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/gen3c_single_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/inference/gen3c_single_image.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/gen3c_single_image_sdg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/inference/gen3c_single_image_sdg.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/inference_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/inference/inference_utils.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/ray_triangle_intersection_warp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/inference/ray_triangle_intersection_warp.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/text2world.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/inference/text2world.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/text2world_multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/inference/text2world_multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/video2world.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/inference/video2world.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/video2world_multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/inference/video2world_multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/vipe_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/inference/vipe_utils.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/world_generation_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/inference/world_generation_pipeline.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/inference/world_interpolator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/inference/world_interpolator.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/model/bu_model_world_interpolator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/model/bu_model_world_interpolator.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/model/model_gen3c.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/model/model_gen3c.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/model/model_t2w.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/model/model_t2w.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/model/model_t2w_multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/model/model_t2w_multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/model/model_v2w.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/model/model_v2w.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/model/model_v2w_multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/model/model_v2w_multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/model/model_world_interpolator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/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/lyra/HEAD/cosmos_predict1/diffusion/module/attention.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/module/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/module/blocks.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/module/parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/module/parallel.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/module/position_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/module/position_embedding.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/module/pretrained_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/module/pretrained_vae.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/module/timm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/module/timm.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/modules/denoiser_scaling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/modules/denoiser_scaling.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/modules/res_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/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/lyra/HEAD/cosmos_predict1/diffusion/networks/general_dit.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/networks/general_dit_multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/networks/general_dit_multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/networks/general_dit_video_conditioned.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/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/lyra/HEAD/cosmos_predict1/diffusion/networks/general_dit_video_conditioned_multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/prompt_upsampler/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/prompt_upsampler/inference.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/prompt_upsampler/text2world_prompt_upsampler_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/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/lyra/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/lyra/HEAD/cosmos_predict1/diffusion/training/callbacks/every_n.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/callbacks/grad_clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/callbacks/grad_clip.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/callbacks/iter_speed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/callbacks/iter_speed.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/callbacks/low_precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/callbacks/low_precision.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/conditioner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/conditioner.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/base/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/config/base/ema.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/base/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/config/base/model.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/base/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/config/base/optim.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/base/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/config/base/vae.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/config/config.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/config_multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/config/config_multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/text2world/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/config/text2world/experiment.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/text2world/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/config/text2world/registry.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/text2world_multiview/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/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/lyra/HEAD/cosmos_predict1/diffusion/training/config/text2world_multiview/registry.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/video2world/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/config/video2world/experiment.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/video2world/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/config/video2world/registry.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/video2world_action/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/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/lyra/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/lyra/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/lyra/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/lyra/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/lyra/HEAD/cosmos_predict1/diffusion/training/config/video2world_multiview/registry.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/config/world_interpolator/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/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/lyra/HEAD/cosmos_predict1/diffusion/training/config/world_interpolator/registry.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/context_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/context_parallel.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/datasets/data_sources/item_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/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/lyra/HEAD/cosmos_predict1/diffusion/training/datasets/dataset_3D.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/datasets/dataset_3D_binary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/datasets/dataset_3D_binary.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/datasets/dataset_multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/datasets/dataset_multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/datasets/dataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/datasets/dataset_utils.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/datasets/dataset_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/datasets/dataset_video.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/functional/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/functional/loss.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/functional/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/functional/lr_scheduler.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/models/extend_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/models/extend_model.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/models/extend_model_multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/models/extend_model_multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/models/interpolator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/models/interpolator.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/models/model.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/models/model_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/models/model_image.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/models/model_multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/models/model_multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/models/model_peft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/models/model_peft.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/module/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/module/blocks.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/module/position_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/module/position_embedding.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/module/pretrained_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/module/pretrained_vae.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/module/pretrained_vae_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/module/pretrained_vae_base.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/modules/edm_sde.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/modules/edm_sde.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/networks/general_dit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/networks/general_dit.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/networks/general_dit_action.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/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/lyra/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/lyra/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/lyra/HEAD/cosmos_predict1/diffusion/training/networks/general_dit_multiview.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/tensor_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/tensor_parallel.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/train.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/trainer.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/utils/checkpointer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/utils/checkpointer.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/utils/fsdp_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/utils/fsdp_helper.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/utils/inference_long_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/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/lyra/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/lyra/HEAD/cosmos_predict1/diffusion/training/utils/optim_instantiate.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/utils/peft/lora_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/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/lyra/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/lyra/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/lyra/HEAD/cosmos_predict1/diffusion/training/utils/peft/lora_net.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/training/utils/peft/peft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/training/utils/peft/peft.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/types.py -------------------------------------------------------------------------------- /cosmos_predict1/diffusion/utils/customization/customization_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/diffusion/utils/customization/customization_manager.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/inference/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/inference/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/inference/image_cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/inference/image_cli.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/inference/image_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/inference/image_lib.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/inference/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/inference/utils.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/inference/video_cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/inference/video_cli.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/inference/video_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/inference/video_lib.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/modules/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/modules/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/modules/distributions.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/modules/layers2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/modules/layers2d.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/modules/layers2d_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/modules/layers2d_test.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/modules/layers3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/modules/layers3d.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/modules/layers3d_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/modules/layers3d_test.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/modules/patching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/modules/patching.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/modules/quantizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/modules/quantizers.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/modules/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/modules/utils.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/networks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/networks/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/networks/configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/networks/configs.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/networks/continuous_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/networks/continuous_image.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/networks/continuous_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/networks/continuous_video.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/networks/discrete_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/networks/discrete_image.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/networks/discrete_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/networks/discrete_video.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/test_data/video.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/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/lyra/HEAD/cosmos_predict1/tokenizer/training/callbacks.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/checkpointer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/training/checkpointer.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/configs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/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/lyra/HEAD/cosmos_predict1/tokenizer/training/configs/base/callback.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/configs/base/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/training/configs/base/checkpoint.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/configs/base/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/training/configs/base/data.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/configs/base/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/training/configs/base/loss.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/configs/base/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/training/configs/base/metric.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/configs/base/mock_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/training/configs/base/mock_data.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/configs/base/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/training/configs/base/model.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/configs/base/net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/training/configs/base/net.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/configs/base/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/training/configs/base/optim.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/configs/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/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/lyra/HEAD/cosmos_predict1/tokenizer/training/configs/experiments/basic.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/configs/experiments/cosmos_tokenize1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/training/configs/experiments/cosmos_tokenize1.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/configs/experiments/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/training/configs/experiments/utils.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/configs/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/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/lyra/HEAD/cosmos_predict1/tokenizer/training/datasets/augmentation_provider.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/datasets/augmentors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/training/datasets/augmentors.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/datasets/dataset_provider.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/training/datasets/dataset_provider.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/datasets/mock_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/training/datasets/mock_dataset.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/training/datasets/utils.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/datasets/video_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/training/datasets/video_dataset.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/jit_cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/training/jit_cli.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/training/losses/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/losses/continuous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/training/losses/continuous.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/losses/lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/training/losses/lpips.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/training/metrics.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/training/model.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/training/train.py -------------------------------------------------------------------------------- /cosmos_predict1/tokenizer/training/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/tokenizer/training/trainer.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/base_world_generation_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/base_world_generation_pipeline.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/callback.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/callbacks/grad_clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/callbacks/grad_clip.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/checkpointer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/checkpointer.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/config.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/config_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/config_helper.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/ddp_checkpointer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/ddp_checkpointer.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/device.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/device.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/distributed.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/easy_io/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/backends/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/easy_io/backends/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/backends/base_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/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/lyra/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/lyra/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/lyra/HEAD/cosmos_predict1/utils/easy_io/backends/registry_utils.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/easy_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/easy_io/easy_io.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/file_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/easy_io/file_client.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/handlers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/easy_io/handlers/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/handlers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/easy_io/handlers/base.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/easy_io/handlers/csv_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/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/lyra/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/lyra/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/lyra/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/lyra/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/lyra/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/lyra/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/lyra/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/lyra/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/lyra/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/lyra/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/lyra/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/lyra/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/lyra/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/lyra/HEAD/cosmos_predict1/utils/easy_io/handlers/yaml_handler.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/ema.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/env_parsers/cred_env_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/env_parsers/cred_env_parser.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/env_parsers/env_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/env_parsers/env_parser.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/fsdp_checkpointer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/fsdp_checkpointer.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/fsdp_optim_fix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/fsdp_optim_fix.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/fused_adam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/fused_adam.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/io.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/lazy_config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/lazy_config/__init__.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/lazy_config/file_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/lazy_config/file_io.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/lazy_config/instantiate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/lazy_config/instantiate.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/lazy_config/lazy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/lazy_config/lazy.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/lazy_config/omegaconf_patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/lazy_config/omegaconf_patch.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/lazy_config/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/lazy_config/registry.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/log.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/misc.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/model.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/parallel_state_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/parallel_state_helper.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/scheduler.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/trainer.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/validator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/validator.py -------------------------------------------------------------------------------- /cosmos_predict1/utils/visualize/video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/cosmos_predict1/utils/visualize/video.py -------------------------------------------------------------------------------- /inference.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/inference.sh -------------------------------------------------------------------------------- /lyra.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/lyra.yaml -------------------------------------------------------------------------------- /requirements_gen3c.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/requirements_gen3c.txt -------------------------------------------------------------------------------- /requirements_lyra.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/requirements_lyra.txt -------------------------------------------------------------------------------- /sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/sample.py -------------------------------------------------------------------------------- /scripts/bash/dynamic_sdg.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/scripts/bash/dynamic_sdg.sh -------------------------------------------------------------------------------- /scripts/bash/static_sdg.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/scripts/bash/static_sdg.sh -------------------------------------------------------------------------------- /scripts/download_gen3c_checkpoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/scripts/download_gen3c_checkpoints.py -------------------------------------------------------------------------------- /scripts/download_guardrail_checkpoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/scripts/download_guardrail_checkpoints.py -------------------------------------------------------------------------------- /scripts/download_lyra_checkpoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/scripts/download_lyra_checkpoints.py -------------------------------------------------------------------------------- /scripts/download_tokenizer_checkpoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/scripts/download_tokenizer_checkpoints.py -------------------------------------------------------------------------------- /scripts/test_environment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/scripts/test_environment.py -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/__init__.py -------------------------------------------------------------------------------- /src/eval/compute_metrics_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/eval/compute_metrics_datasets.py -------------------------------------------------------------------------------- /src/eval/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/eval/metrics.py -------------------------------------------------------------------------------- /src/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/models/__init__.py -------------------------------------------------------------------------------- /src/models/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/models/data/__init__.py -------------------------------------------------------------------------------- /src/models/data/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/models/data/base.py -------------------------------------------------------------------------------- /src/models/data/datafield.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/models/data/datafield.py -------------------------------------------------------------------------------- /src/models/data/provider.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/models/data/provider.py -------------------------------------------------------------------------------- /src/models/data/radym.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/models/data/radym.py -------------------------------------------------------------------------------- /src/models/data/radym_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/models/data/radym_wrapper.py -------------------------------------------------------------------------------- /src/models/data/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/models/data/registry.py -------------------------------------------------------------------------------- /src/models/recon/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/models/recon/__init__.py -------------------------------------------------------------------------------- /src/models/recon/model_latent_recon.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/models/recon/model_latent_recon.py -------------------------------------------------------------------------------- /src/models/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/models/utils/__init__.py -------------------------------------------------------------------------------- /src/models/utils/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/models/utils/attention.py -------------------------------------------------------------------------------- /src/models/utils/cosmos_1_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/models/utils/cosmos_1_tokenizer.py -------------------------------------------------------------------------------- /src/models/utils/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/models/utils/data.py -------------------------------------------------------------------------------- /src/models/utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/models/utils/loss.py -------------------------------------------------------------------------------- /src/models/utils/mamba2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/models/utils/mamba2.py -------------------------------------------------------------------------------- /src/models/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/models/utils/misc.py -------------------------------------------------------------------------------- /src/models/utils/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/models/utils/model.py -------------------------------------------------------------------------------- /src/models/utils/render.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/models/utils/render.py -------------------------------------------------------------------------------- /src/models/utils/token_pruning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/models/utils/token_pruning.py -------------------------------------------------------------------------------- /src/models/utils/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/models/utils/train.py -------------------------------------------------------------------------------- /src/rendering/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/rendering/__init__.py -------------------------------------------------------------------------------- /src/rendering/gs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/rendering/gs.py -------------------------------------------------------------------------------- /src/rendering/gs_deferred.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/rendering/gs_deferred.py -------------------------------------------------------------------------------- /src/rendering/gs_deferred_patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/rendering/gs_deferred_patch.py -------------------------------------------------------------------------------- /src/utils/copy_to_resume.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/utils/copy_to_resume.py -------------------------------------------------------------------------------- /src/utils/random_state_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/utils/random_state_utils.py -------------------------------------------------------------------------------- /src/utils/visu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/utils/visu.py -------------------------------------------------------------------------------- /src/visu/dynamic_grid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/visu/dynamic_grid.py -------------------------------------------------------------------------------- /src/visu/dynamic_scene_merge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/visu/dynamic_scene_merge.py -------------------------------------------------------------------------------- /src/visu/dynamic_scene_video_scene_and_camera.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/visu/dynamic_scene_video_scene_and_camera.py -------------------------------------------------------------------------------- /src/visu/dynamic_scene_video_static_camera.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/visu/dynamic_scene_video_static_camera.py -------------------------------------------------------------------------------- /src/visu/dynamic_scene_video_static_scene.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/visu/dynamic_scene_video_static_scene.py -------------------------------------------------------------------------------- /src/visu/dynamic_visu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/visu/dynamic_visu.sh -------------------------------------------------------------------------------- /src/visu/static_visu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/src/visu/static_visu.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/train.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nv-tlabs/lyra/HEAD/train.sh --------------------------------------------------------------------------------