├── .coveragerc ├── .dockerignore ├── .gitattributes ├── .githooks └── submodule-check.sh ├── .github ├── ISSUE_TEMPLATE │ ├── bug.md │ ├── documentation.md │ ├── feature.md │ └── question.md ├── PULL_REQUEST_TEMPLATE.md └── workflows │ └── release-from-tag.yml ├── .gitignore ├── .gitlab-ci.yml ├── .gitlab ├── issue_templates │ ├── bug.md │ ├── documentation.md │ ├── feature.md │ └── question.md ├── merge_request_templates │ └── Default.md ├── scripts │ ├── nvcf_helm_cleanup.sh │ ├── nvcf_helm_deploy.sh │ ├── nvcf_helm_validate.sh │ ├── nvcf_split_benchmark.sh │ ├── run_env_tests.sh │ └── slurm_end_to_end.sh └── vault │ ├── ci-secrets.tmpl │ ├── vault-agent.config │ └── vault-auth.sh ├── .gitmodules ├── .pre-commit-config.yaml ├── AGENTS.md ├── ATTRIBUTIONS.md ├── CHANGELOG.md ├── CLAUDE.md ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── GEMINI.md ├── ISSUES.md ├── LICENSE ├── MODEL_CARDS.md ├── README.md ├── RESPONSIBLE_USE.md ├── SECURITY.md ├── benchmarks ├── __init__.py ├── cloudevent.py ├── secrets.py ├── split_pipeline │ ├── __init__.py │ ├── deploy.json │ ├── invoke.json │ └── nvcf_split_benchmark.py └── summary.py ├── charts └── cosmos-curate │ ├── .helmignore │ ├── Chart.lock │ ├── Chart.yaml │ ├── README.md │ ├── files │ └── extractor.py │ ├── templates │ ├── _helpers.tpl │ ├── curate-configmap.yaml │ ├── image-pull-secret.yaml │ ├── ngc-secret.yaml │ ├── otel-collector-config.yaml │ ├── s3-secret.yaml │ ├── secret-extractor-configmap.yaml │ ├── service.yaml │ ├── servicemonitor.yaml │ └── statefulset.yaml │ ├── values-standalone.yaml │ └── values.yaml ├── cosmos_curate ├── __init__.py ├── client │ ├── __init__.py │ ├── cli.py │ ├── environment.pyi │ ├── image_cli │ │ ├── __init__.py │ │ └── image_app.py │ ├── local_cli │ │ ├── __init__.py │ │ └── launch_local.py │ ├── nvcf_cli │ │ ├── __init__.py │ │ ├── launch_nvcf.py │ │ └── ncf │ │ │ ├── __init__.py │ │ │ ├── asset │ │ │ ├── __init__.py │ │ │ └── asset_manager.py │ │ │ ├── common │ │ │ ├── __init__.py │ │ │ ├── errors.py │ │ │ ├── nvcf_base.py │ │ │ └── nvcf_client.py │ │ │ ├── config │ │ │ ├── __init__.py │ │ │ └── config_manager.py │ │ │ ├── image │ │ │ ├── __init__.py │ │ │ └── image_manager.py │ │ │ ├── launcher │ │ │ ├── __init__.py │ │ │ ├── nvcf_driver.py │ │ │ ├── nvcf_function.py │ │ │ └── nvcf_helper.py │ │ │ └── model │ │ │ ├── __init__.py │ │ │ └── model_manager.py │ ├── slurm_cli │ │ ├── __init__.py │ │ ├── prometheus_service_discovery.py │ │ ├── sbatch.sh.j2 │ │ └── slurm.py │ ├── utils │ │ ├── __init__.py │ │ ├── conda_envs.py │ │ ├── docker_utils.py │ │ └── validations.py │ └── view_cli │ │ ├── __init__.py │ │ ├── clip_viewer.py │ │ ├── clip_viewer_webdocs │ │ ├── images │ │ │ └── nvidia-logo.webp │ │ ├── index.html │ │ ├── script.js │ │ └── style.css │ │ └── viewer_app.py ├── configs │ └── all_models.json ├── core │ ├── __init__.py │ ├── cf │ │ ├── README.md │ │ ├── __init__.py │ │ ├── nvcf_main.py │ │ ├── nvcf_model_manager.py │ │ └── nvcf_utils.py │ ├── interfaces │ │ ├── __init__.py │ │ ├── model_interface.py │ │ ├── pipeline_interface.py │ │ ├── runner_interface.py │ │ └── stage_interface.py │ ├── managers │ │ ├── __init__.py │ │ ├── model_cli.py │ │ └── postgres_cli.py │ └── utils │ │ ├── __init__.py │ │ ├── config │ │ ├── __init__.py │ │ ├── args_utils.py │ │ ├── config.py │ │ └── operation_context.py │ │ ├── dataset │ │ ├── __init__.py │ │ ├── dimensions.py │ │ └── webdataset_utils.py │ │ ├── db │ │ ├── __init__.py │ │ ├── database_types.py │ │ ├── database_utils.py │ │ └── postgres_schema_types.py │ │ ├── environment.py │ │ ├── infra │ │ ├── __init__.py │ │ ├── gpu_start_helper.py │ │ ├── hardware_info.py │ │ ├── performance_utils.py │ │ └── ray_cluster_utils.py │ │ ├── misc │ │ ├── __init__.py │ │ ├── file_lock.py │ │ ├── grouping.py │ │ ├── retry_utils.py │ │ └── summarize.py │ │ ├── model │ │ ├── __init__.py │ │ ├── conda_utils.py │ │ └── model_utils.py │ │ ├── pixi_runtime_envs.py │ │ └── storage │ │ ├── __init__.py │ │ ├── azure_client.py │ │ ├── presigned_s3_zip.py │ │ ├── s3_client.py │ │ ├── storage_client.py │ │ ├── storage_utils.py │ │ └── writer_utils.py ├── models │ ├── README.md │ ├── __init__.py │ ├── aesthetics.py │ ├── all_models.py │ ├── chat_lm.py │ ├── clip.py │ ├── clip_aesthetics.py │ ├── configs │ │ ├── internvideo2_mm_config_bert.json │ │ ├── internvideo2_mm_config_model.json │ │ └── t5_encoder.json │ ├── cosmos_embed1.py │ ├── gpt2.py │ ├── internvideo2_mm.py │ ├── internvideo2_multi_modality │ │ ├── bert │ │ │ ├── __init__.py │ │ │ ├── builder.py │ │ │ ├── tokenization_bert.py │ │ │ └── xbert.py │ │ └── internvideo2 │ │ │ ├── __init__.py │ │ │ ├── flash_attention_class.py │ │ │ ├── internvideo2.py │ │ │ └── pos_embed.py │ ├── prompts.py │ ├── qwen_vl.py │ ├── t5_encoder.py │ ├── transnetv2.py │ ├── vllm_cosmos_reason1_vl.py │ ├── vllm_interface.py │ ├── vllm_model_ids.py │ ├── vllm_nemotron.py │ ├── vllm_phi.py │ ├── vllm_plugin.py │ └── vllm_qwen.py ├── pipelines │ ├── README.md │ ├── __init__.py │ ├── av │ │ ├── __init__.py │ │ ├── av_video_captioning_pipeline.py │ │ ├── av_video_ingesting_pipeline.py │ │ ├── av_video_pipelines_common.py │ │ ├── av_video_sharding_pipeline.py │ │ ├── av_video_splitting_pipeline.py │ │ ├── captioning │ │ │ ├── __init__.py │ │ │ └── captioning_stages.py │ │ ├── clipping │ │ │ ├── __init__.py │ │ │ └── clip_extraction_stages.py │ │ ├── downloaders │ │ │ ├── __init__.py │ │ │ └── download_stages.py │ │ ├── pipeline_args.py │ │ ├── run_pipeline.py │ │ ├── trajectory │ │ │ └── __init__.py │ │ ├── utils │ │ │ ├── __init__.py │ │ │ ├── av_data_info.py │ │ │ ├── av_data_model.py │ │ │ ├── av_pipe_input.py │ │ │ ├── postgres_schema.py │ │ │ └── run_utils.py │ │ └── writers │ │ │ ├── __init__.py │ │ │ ├── annotation_writer_stage.py │ │ │ ├── base_writer_stage.py │ │ │ ├── clip_writer_stage.py │ │ │ ├── cosmos_predict2_writer_stage.py │ │ │ ├── dataset_writer_stage.py │ │ │ ├── make_db_row.py │ │ │ ├── t5_writer_stage.py │ │ │ └── trajectory_writer_stage.py │ ├── examples │ │ ├── __init__.py │ │ ├── demo_task_chunking_pipeline.py │ │ ├── example_captioning_stages.py │ │ └── hello_world_pipeline.py │ ├── pipeline_args.py │ └── video │ │ ├── __init__.py │ │ ├── captioning │ │ ├── __init__.py │ │ ├── api_caption_stage.py │ │ ├── captioning_stages.py │ │ └── vllm_caption_stage.py │ │ ├── clipping │ │ ├── __init__.py │ │ ├── clip_extraction_stages.py │ │ ├── clip_frame_extraction_stages.py │ │ ├── frame_extraction_stages.py │ │ └── transnetv2_extraction_stages.py │ │ ├── dedup │ │ ├── __init__.py │ │ ├── dedup_actor.py │ │ └── raft_actor.py │ │ ├── dedup_pipeline.py │ │ ├── embedding │ │ ├── __init__.py │ │ ├── cosmos_embed1_stages.py │ │ └── internvideo2_stages.py │ │ ├── filtering │ │ ├── __init__.py │ │ ├── aesthetics │ │ │ ├── __init__.py │ │ │ ├── aesthetic_filter_stages.py │ │ │ └── qwen_filter_stages.py │ │ └── motion │ │ │ ├── __init__.py │ │ │ ├── motion_filter_stages.py │ │ │ └── motion_vector_backend.py │ │ ├── preview │ │ ├── __init__.py │ │ └── preview_stages.py │ │ ├── read_write │ │ ├── __init__.py │ │ ├── download_stages.py │ │ ├── metadata_writer_stage.py │ │ ├── remux_stages.py │ │ └── summary_writers.py │ │ ├── run_pipeline.py │ │ ├── sharding_pipeline.py │ │ ├── splitting_pipeline.py │ │ └── utils │ │ ├── __init__.py │ │ ├── data_model.py │ │ ├── decoder_utils.py │ │ ├── nvcodec_utils.py │ │ ├── video_pipe_input.py │ │ ├── vision_process.py │ │ └── windowing_utils.py └── scripts │ ├── __init__.py │ ├── onto_nvcf.py │ └── onto_slurm.py ├── devset.sh ├── docs ├── DEVELOPER_GUIDE.md ├── README.md ├── assets │ ├── cosmos-curate-dashboard.png │ ├── cosmos-curate-logical-view.png │ ├── cosmos-curate-physical-view.png │ ├── cosmos-curate-pipelines.png │ └── nvidia-cosmos-header.png ├── client │ ├── END_USER_GUIDE.md │ └── NVCF_GUIDE.md └── curator │ ├── ARCHITECTURE_GUIDE.md │ ├── OBSERVABILITY_GUIDE.md │ ├── PIPELINE_DESIGN_GUIDE.md │ ├── REFERENCE_PIPELINES_AV.md │ ├── REFERENCE_PIPELINES_VIDEO.md │ ├── VLLM_INTERFACE_DEBUG.md │ ├── VLLM_INTERFACE_DESIGN.md │ └── VLLM_INTERFACE_PLUGIN.md ├── examples ├── nvcf │ ├── .gitignore │ ├── ci │ │ ├── ci_create_helm_default.json.template │ │ └── ci_deploy_helm_default.json.template │ ├── creds │ │ ├── aws_credentials │ │ └── s3cred-function.json │ ├── function │ │ ├── create_curator_helm.json │ │ ├── deploy_curator_helm.json │ │ ├── invoke_video_dedup.json │ │ ├── invoke_video_shard.json │ │ ├── invoke_video_split.json │ │ └── invoke_video_split_full.json │ ├── image │ │ └── image_upload.json │ └── model │ │ ├── model_upload_aesthetic.json │ │ ├── model_upload_bert.json │ │ ├── model_upload_gemma_vllm.json │ │ ├── model_upload_internvideo2.json │ │ ├── model_upload_kosmos2.json │ │ ├── model_upload_llava_mistral.json │ │ ├── model_upload_openai_clip.json │ │ ├── model_upload_qwen_lm.json │ │ ├── model_upload_qwen_vl.json │ │ ├── model_upload_t5_xxl.json │ │ └── model_upload_transnetv2.json ├── observability │ └── grafana │ │ └── cosmos-curate-oss.json ├── slurm │ ├── ci_run_end_to_end.sh │ ├── source_me_env_vars.sh │ ├── source_me_source_code_mount.sh │ ├── sync_config_creds_models.sh │ ├── sync_source_code.sh │ └── upload_image.sh └── workflow │ ├── .gitignore │ ├── example_input_file.json │ ├── example_nvcf_func_file.json │ ├── operate_nvcf_functions.py │ └── template_invoke_video_split.json ├── package ├── README.md └── cosmos_curate │ ├── default.dockerfile.jinja2 │ └── envs │ └── paddle-ocr │ └── post_install.sh ├── pixi.lock ├── pixi.toml ├── poetry.lock ├── pyproject.toml ├── pytest.ini ├── setup.py └── tests ├── README.md ├── __init__.py ├── benchmarks ├── conftest.py ├── split_pipeline │ └── test_nvcf_split_benchmark.py ├── test_cloudevent.py ├── test_secrets.py └── test_summary.py ├── cosmos_curate ├── client │ ├── local_cli │ │ └── test_launch_local.py │ ├── nvcf_cli │ │ └── ncf │ │ │ ├── asset │ │ │ └── test_asset_manager.py │ │ │ ├── common │ │ │ ├── test_nvcf_base.py │ │ │ └── test_nvcf_client.py │ │ │ ├── config │ │ │ └── test_config_manager.py │ │ │ ├── image │ │ │ └── test_image_manager.py │ │ │ ├── launcher │ │ │ ├── test_nvcf_driver.py │ │ │ ├── test_nvcf_function.py │ │ │ └── test_nvcf_helper.py │ │ │ └── model │ │ │ └── test_model_manager.py │ ├── slurm_cli │ │ ├── conftest.py │ │ ├── test_prometheus_service_discovery.py │ │ └── test_slurm.py │ └── utils │ │ └── test_validations.py ├── core │ ├── cf │ │ └── test_nvcf_main.py │ ├── managers │ │ ├── test_model_cli.py │ │ └── test_postgres_cli.py │ └── utils │ │ ├── dataset │ │ └── test_dimensions.py │ │ ├── db │ │ └── test_database_utils.py │ │ ├── misc │ │ └── test_grouping.py │ │ ├── model │ │ └── test_model_utils.py │ │ ├── storage │ │ ├── test_presigned_s3_zip.py │ │ └── test_storage_utils.py │ │ └── test_s3_prefix.py ├── models │ ├── test_gpt2.py │ ├── test_prompts.py │ ├── test_vllm_cosmos_reason1_vl.py │ ├── test_vllm_interface.py │ ├── test_vllm_model_ids.py │ ├── test_vllm_nemotron.py │ ├── test_vllm_phi.py │ └── test_vllm_qwen.py ├── pipelines │ ├── av │ │ ├── captioning │ │ │ └── test_captioning_stages.py │ │ ├── clipping │ │ │ └── test_clip_extraction_stages.py │ │ ├── conftest.py │ │ ├── downloaders │ │ │ └── test_download_stages.py │ │ ├── utils │ │ │ ├── test_av_data_model.py │ │ │ └── test_av_pipe_input.py │ │ └── writers │ │ │ ├── test_annotation_writer_stage.py │ │ │ ├── test_clip_writer_stage.py │ │ │ ├── test_cosmos_predict2_writer_stage.py │ │ │ ├── test_dataset_writer_stage.py │ │ │ ├── test_make_db_row.py │ │ │ ├── test_t5_writer_stage.py │ │ │ └── test_utils.py │ ├── conftest.py │ └── video │ │ ├── captioning │ │ ├── example_t5_embeddings.pickle │ │ ├── test_api_caption_stage.py │ │ ├── test_enhance_captions.py │ │ ├── test_qwen_captioning.py │ │ ├── test_t5_embedding.py │ │ └── test_vllm_caption_stage.py │ │ ├── clipping │ │ ├── test_fixed_stride_extraction.py │ │ └── test_transnetv2_extraction.py │ │ ├── conftest.py │ │ ├── embedding │ │ ├── test_cosmos_embed1_embedding.py │ │ └── test_internvideo2_embedding.py │ │ ├── filtering │ │ ├── aesthetics │ │ │ ├── test_aesthetic_filter.py │ │ │ └── test_qwen_filter.py │ │ └── motion │ │ │ └── test_motion_filter.py │ │ ├── read_write │ │ ├── test_metadata_writer_stage.py │ │ └── test_remux_stages.py │ │ └── utils │ │ ├── test_data_model.py │ │ └── test_decoder_utils.py └── scripts │ └── test_onto_slurm.py └── utils ├── __init__.py ├── gpu_worker_limiter.py ├── sequential_runner.py └── vllm_mock.py /.coveragerc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/.coveragerc -------------------------------------------------------------------------------- /.dockerignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/.dockerignore -------------------------------------------------------------------------------- /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/.gitattributes -------------------------------------------------------------------------------- /.githooks/submodule-check.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/.githooks/submodule-check.sh -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/bug.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/.github/ISSUE_TEMPLATE/bug.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/documentation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/.github/ISSUE_TEMPLATE/documentation.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/.github/ISSUE_TEMPLATE/feature.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/question.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/.github/ISSUE_TEMPLATE/question.md -------------------------------------------------------------------------------- /.github/PULL_REQUEST_TEMPLATE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/.github/PULL_REQUEST_TEMPLATE.md -------------------------------------------------------------------------------- /.github/workflows/release-from-tag.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/.github/workflows/release-from-tag.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitlab-ci.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/.gitlab-ci.yml -------------------------------------------------------------------------------- /.gitlab/issue_templates/bug.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/.gitlab/issue_templates/bug.md -------------------------------------------------------------------------------- /.gitlab/issue_templates/documentation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/.gitlab/issue_templates/documentation.md -------------------------------------------------------------------------------- /.gitlab/issue_templates/feature.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/.gitlab/issue_templates/feature.md -------------------------------------------------------------------------------- /.gitlab/issue_templates/question.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/.gitlab/issue_templates/question.md -------------------------------------------------------------------------------- /.gitlab/merge_request_templates/Default.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/.gitlab/merge_request_templates/Default.md -------------------------------------------------------------------------------- /.gitlab/scripts/nvcf_helm_cleanup.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/.gitlab/scripts/nvcf_helm_cleanup.sh -------------------------------------------------------------------------------- /.gitlab/scripts/nvcf_helm_deploy.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/.gitlab/scripts/nvcf_helm_deploy.sh -------------------------------------------------------------------------------- /.gitlab/scripts/nvcf_helm_validate.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/.gitlab/scripts/nvcf_helm_validate.sh -------------------------------------------------------------------------------- /.gitlab/scripts/nvcf_split_benchmark.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/.gitlab/scripts/nvcf_split_benchmark.sh -------------------------------------------------------------------------------- /.gitlab/scripts/run_env_tests.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/.gitlab/scripts/run_env_tests.sh -------------------------------------------------------------------------------- /.gitlab/scripts/slurm_end_to_end.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/.gitlab/scripts/slurm_end_to_end.sh -------------------------------------------------------------------------------- /.gitlab/vault/ci-secrets.tmpl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/.gitlab/vault/ci-secrets.tmpl -------------------------------------------------------------------------------- /.gitlab/vault/vault-agent.config: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/.gitlab/vault/vault-agent.config -------------------------------------------------------------------------------- /.gitlab/vault/vault-auth.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/.gitlab/vault/vault-auth.sh -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/.gitmodules -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /AGENTS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/AGENTS.md -------------------------------------------------------------------------------- /ATTRIBUTIONS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/ATTRIBUTIONS.md -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /CLAUDE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/CLAUDE.md -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /GEMINI.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/GEMINI.md -------------------------------------------------------------------------------- /ISSUES.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/ISSUES.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/LICENSE -------------------------------------------------------------------------------- /MODEL_CARDS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/MODEL_CARDS.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/README.md -------------------------------------------------------------------------------- /RESPONSIBLE_USE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/RESPONSIBLE_USE.md -------------------------------------------------------------------------------- /SECURITY.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/SECURITY.md -------------------------------------------------------------------------------- /benchmarks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/benchmarks/__init__.py -------------------------------------------------------------------------------- /benchmarks/cloudevent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/benchmarks/cloudevent.py -------------------------------------------------------------------------------- /benchmarks/secrets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/benchmarks/secrets.py -------------------------------------------------------------------------------- /benchmarks/split_pipeline/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/benchmarks/split_pipeline/__init__.py -------------------------------------------------------------------------------- /benchmarks/split_pipeline/deploy.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/benchmarks/split_pipeline/deploy.json -------------------------------------------------------------------------------- /benchmarks/split_pipeline/invoke.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/benchmarks/split_pipeline/invoke.json -------------------------------------------------------------------------------- /benchmarks/split_pipeline/nvcf_split_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/benchmarks/split_pipeline/nvcf_split_benchmark.py -------------------------------------------------------------------------------- /benchmarks/summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/benchmarks/summary.py -------------------------------------------------------------------------------- /charts/cosmos-curate/.helmignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/charts/cosmos-curate/.helmignore -------------------------------------------------------------------------------- /charts/cosmos-curate/Chart.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/charts/cosmos-curate/Chart.lock -------------------------------------------------------------------------------- /charts/cosmos-curate/Chart.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/charts/cosmos-curate/Chart.yaml -------------------------------------------------------------------------------- /charts/cosmos-curate/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/charts/cosmos-curate/README.md -------------------------------------------------------------------------------- /charts/cosmos-curate/files/extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/charts/cosmos-curate/files/extractor.py -------------------------------------------------------------------------------- /charts/cosmos-curate/templates/_helpers.tpl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/charts/cosmos-curate/templates/_helpers.tpl -------------------------------------------------------------------------------- /charts/cosmos-curate/templates/curate-configmap.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/charts/cosmos-curate/templates/curate-configmap.yaml -------------------------------------------------------------------------------- /charts/cosmos-curate/templates/image-pull-secret.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/charts/cosmos-curate/templates/image-pull-secret.yaml -------------------------------------------------------------------------------- /charts/cosmos-curate/templates/ngc-secret.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/charts/cosmos-curate/templates/ngc-secret.yaml -------------------------------------------------------------------------------- /charts/cosmos-curate/templates/otel-collector-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/charts/cosmos-curate/templates/otel-collector-config.yaml -------------------------------------------------------------------------------- /charts/cosmos-curate/templates/s3-secret.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/charts/cosmos-curate/templates/s3-secret.yaml -------------------------------------------------------------------------------- /charts/cosmos-curate/templates/secret-extractor-configmap.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/charts/cosmos-curate/templates/secret-extractor-configmap.yaml -------------------------------------------------------------------------------- /charts/cosmos-curate/templates/service.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/charts/cosmos-curate/templates/service.yaml -------------------------------------------------------------------------------- /charts/cosmos-curate/templates/servicemonitor.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/charts/cosmos-curate/templates/servicemonitor.yaml -------------------------------------------------------------------------------- /charts/cosmos-curate/templates/statefulset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/charts/cosmos-curate/templates/statefulset.yaml -------------------------------------------------------------------------------- /charts/cosmos-curate/values-standalone.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/charts/cosmos-curate/values-standalone.yaml -------------------------------------------------------------------------------- /charts/cosmos-curate/values.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/charts/cosmos-curate/values.yaml -------------------------------------------------------------------------------- /cosmos_curate/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/client/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/client/cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/cli.py -------------------------------------------------------------------------------- /cosmos_curate/client/environment.pyi: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/environment.pyi -------------------------------------------------------------------------------- /cosmos_curate/client/image_cli/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/image_cli/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/client/image_cli/image_app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/image_cli/image_app.py -------------------------------------------------------------------------------- /cosmos_curate/client/local_cli/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/local_cli/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/client/local_cli/launch_local.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/local_cli/launch_local.py -------------------------------------------------------------------------------- /cosmos_curate/client/nvcf_cli/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/nvcf_cli/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/client/nvcf_cli/launch_nvcf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/nvcf_cli/launch_nvcf.py -------------------------------------------------------------------------------- /cosmos_curate/client/nvcf_cli/ncf/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/nvcf_cli/ncf/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/client/nvcf_cli/ncf/asset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/nvcf_cli/ncf/asset/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/client/nvcf_cli/ncf/asset/asset_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/nvcf_cli/ncf/asset/asset_manager.py -------------------------------------------------------------------------------- /cosmos_curate/client/nvcf_cli/ncf/common/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/nvcf_cli/ncf/common/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/client/nvcf_cli/ncf/common/errors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/nvcf_cli/ncf/common/errors.py -------------------------------------------------------------------------------- /cosmos_curate/client/nvcf_cli/ncf/common/nvcf_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/nvcf_cli/ncf/common/nvcf_base.py -------------------------------------------------------------------------------- /cosmos_curate/client/nvcf_cli/ncf/common/nvcf_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/nvcf_cli/ncf/common/nvcf_client.py -------------------------------------------------------------------------------- /cosmos_curate/client/nvcf_cli/ncf/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/nvcf_cli/ncf/config/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/client/nvcf_cli/ncf/config/config_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/nvcf_cli/ncf/config/config_manager.py -------------------------------------------------------------------------------- /cosmos_curate/client/nvcf_cli/ncf/image/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/nvcf_cli/ncf/image/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/client/nvcf_cli/ncf/image/image_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/nvcf_cli/ncf/image/image_manager.py -------------------------------------------------------------------------------- /cosmos_curate/client/nvcf_cli/ncf/launcher/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/nvcf_cli/ncf/launcher/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/client/nvcf_cli/ncf/launcher/nvcf_driver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/nvcf_cli/ncf/launcher/nvcf_driver.py -------------------------------------------------------------------------------- /cosmos_curate/client/nvcf_cli/ncf/launcher/nvcf_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/nvcf_cli/ncf/launcher/nvcf_function.py -------------------------------------------------------------------------------- /cosmos_curate/client/nvcf_cli/ncf/launcher/nvcf_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/nvcf_cli/ncf/launcher/nvcf_helper.py -------------------------------------------------------------------------------- /cosmos_curate/client/nvcf_cli/ncf/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/nvcf_cli/ncf/model/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/client/nvcf_cli/ncf/model/model_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/nvcf_cli/ncf/model/model_manager.py -------------------------------------------------------------------------------- /cosmos_curate/client/slurm_cli/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/slurm_cli/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/client/slurm_cli/prometheus_service_discovery.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/slurm_cli/prometheus_service_discovery.py -------------------------------------------------------------------------------- /cosmos_curate/client/slurm_cli/sbatch.sh.j2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/slurm_cli/sbatch.sh.j2 -------------------------------------------------------------------------------- /cosmos_curate/client/slurm_cli/slurm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/slurm_cli/slurm.py -------------------------------------------------------------------------------- /cosmos_curate/client/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/utils/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/client/utils/conda_envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/utils/conda_envs.py -------------------------------------------------------------------------------- /cosmos_curate/client/utils/docker_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/utils/docker_utils.py -------------------------------------------------------------------------------- /cosmos_curate/client/utils/validations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/utils/validations.py -------------------------------------------------------------------------------- /cosmos_curate/client/view_cli/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/view_cli/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/client/view_cli/clip_viewer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/view_cli/clip_viewer.py -------------------------------------------------------------------------------- /cosmos_curate/client/view_cli/clip_viewer_webdocs/images/nvidia-logo.webp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/view_cli/clip_viewer_webdocs/images/nvidia-logo.webp -------------------------------------------------------------------------------- /cosmos_curate/client/view_cli/clip_viewer_webdocs/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/view_cli/clip_viewer_webdocs/index.html -------------------------------------------------------------------------------- /cosmos_curate/client/view_cli/clip_viewer_webdocs/script.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/view_cli/clip_viewer_webdocs/script.js -------------------------------------------------------------------------------- /cosmos_curate/client/view_cli/clip_viewer_webdocs/style.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/view_cli/clip_viewer_webdocs/style.css -------------------------------------------------------------------------------- /cosmos_curate/client/view_cli/viewer_app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/client/view_cli/viewer_app.py -------------------------------------------------------------------------------- /cosmos_curate/configs/all_models.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/configs/all_models.json -------------------------------------------------------------------------------- /cosmos_curate/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/core/cf/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/cf/README.md -------------------------------------------------------------------------------- /cosmos_curate/core/cf/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/cf/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/core/cf/nvcf_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/cf/nvcf_main.py -------------------------------------------------------------------------------- /cosmos_curate/core/cf/nvcf_model_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/cf/nvcf_model_manager.py -------------------------------------------------------------------------------- /cosmos_curate/core/cf/nvcf_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/cf/nvcf_utils.py -------------------------------------------------------------------------------- /cosmos_curate/core/interfaces/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/interfaces/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/core/interfaces/model_interface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/interfaces/model_interface.py -------------------------------------------------------------------------------- /cosmos_curate/core/interfaces/pipeline_interface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/interfaces/pipeline_interface.py -------------------------------------------------------------------------------- /cosmos_curate/core/interfaces/runner_interface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/interfaces/runner_interface.py -------------------------------------------------------------------------------- /cosmos_curate/core/interfaces/stage_interface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/interfaces/stage_interface.py -------------------------------------------------------------------------------- /cosmos_curate/core/managers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/managers/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/core/managers/model_cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/managers/model_cli.py -------------------------------------------------------------------------------- /cosmos_curate/core/managers/postgres_cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/managers/postgres_cli.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/config/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/config/args_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/config/args_utils.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/config/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/config/config.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/config/operation_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/config/operation_context.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/dataset/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/dataset/dimensions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/dataset/dimensions.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/dataset/webdataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/dataset/webdataset_utils.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/db/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/db/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/db/database_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/db/database_types.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/db/database_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/db/database_utils.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/db/postgres_schema_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/db/postgres_schema_types.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/environment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/environment.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/infra/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/infra/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/infra/gpu_start_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/infra/gpu_start_helper.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/infra/hardware_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/infra/hardware_info.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/infra/performance_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/infra/performance_utils.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/infra/ray_cluster_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/infra/ray_cluster_utils.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/misc/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/misc/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/misc/file_lock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/misc/file_lock.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/misc/grouping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/misc/grouping.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/misc/retry_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/misc/retry_utils.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/misc/summarize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/misc/summarize.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/model/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/model/conda_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/model/conda_utils.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/model/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/model/model_utils.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/pixi_runtime_envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/pixi_runtime_envs.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/storage/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/storage/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/storage/azure_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/storage/azure_client.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/storage/presigned_s3_zip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/storage/presigned_s3_zip.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/storage/s3_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/storage/s3_client.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/storage/storage_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/storage/storage_client.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/storage/storage_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/storage/storage_utils.py -------------------------------------------------------------------------------- /cosmos_curate/core/utils/storage/writer_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/core/utils/storage/writer_utils.py -------------------------------------------------------------------------------- /cosmos_curate/models/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/README.md -------------------------------------------------------------------------------- /cosmos_curate/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/models/aesthetics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/aesthetics.py -------------------------------------------------------------------------------- /cosmos_curate/models/all_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/all_models.py -------------------------------------------------------------------------------- /cosmos_curate/models/chat_lm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/chat_lm.py -------------------------------------------------------------------------------- /cosmos_curate/models/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/clip.py -------------------------------------------------------------------------------- /cosmos_curate/models/clip_aesthetics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/clip_aesthetics.py -------------------------------------------------------------------------------- /cosmos_curate/models/configs/internvideo2_mm_config_bert.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/configs/internvideo2_mm_config_bert.json -------------------------------------------------------------------------------- /cosmos_curate/models/configs/internvideo2_mm_config_model.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/configs/internvideo2_mm_config_model.json -------------------------------------------------------------------------------- /cosmos_curate/models/configs/t5_encoder.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/configs/t5_encoder.json -------------------------------------------------------------------------------- /cosmos_curate/models/cosmos_embed1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/cosmos_embed1.py -------------------------------------------------------------------------------- /cosmos_curate/models/gpt2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/gpt2.py -------------------------------------------------------------------------------- /cosmos_curate/models/internvideo2_mm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/internvideo2_mm.py -------------------------------------------------------------------------------- /cosmos_curate/models/internvideo2_multi_modality/bert/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/internvideo2_multi_modality/bert/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/models/internvideo2_multi_modality/bert/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/internvideo2_multi_modality/bert/builder.py -------------------------------------------------------------------------------- /cosmos_curate/models/internvideo2_multi_modality/bert/tokenization_bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/internvideo2_multi_modality/bert/tokenization_bert.py -------------------------------------------------------------------------------- /cosmos_curate/models/internvideo2_multi_modality/bert/xbert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/internvideo2_multi_modality/bert/xbert.py -------------------------------------------------------------------------------- /cosmos_curate/models/internvideo2_multi_modality/internvideo2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/internvideo2_multi_modality/internvideo2/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/models/internvideo2_multi_modality/internvideo2/flash_attention_class.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/internvideo2_multi_modality/internvideo2/flash_attention_class.py -------------------------------------------------------------------------------- /cosmos_curate/models/internvideo2_multi_modality/internvideo2/internvideo2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/internvideo2_multi_modality/internvideo2/internvideo2.py -------------------------------------------------------------------------------- /cosmos_curate/models/internvideo2_multi_modality/internvideo2/pos_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/internvideo2_multi_modality/internvideo2/pos_embed.py -------------------------------------------------------------------------------- /cosmos_curate/models/prompts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/prompts.py -------------------------------------------------------------------------------- /cosmos_curate/models/qwen_vl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/qwen_vl.py -------------------------------------------------------------------------------- /cosmos_curate/models/t5_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/t5_encoder.py -------------------------------------------------------------------------------- /cosmos_curate/models/transnetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/transnetv2.py -------------------------------------------------------------------------------- /cosmos_curate/models/vllm_cosmos_reason1_vl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/vllm_cosmos_reason1_vl.py -------------------------------------------------------------------------------- /cosmos_curate/models/vllm_interface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/vllm_interface.py -------------------------------------------------------------------------------- /cosmos_curate/models/vllm_model_ids.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/vllm_model_ids.py -------------------------------------------------------------------------------- /cosmos_curate/models/vllm_nemotron.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/vllm_nemotron.py -------------------------------------------------------------------------------- /cosmos_curate/models/vllm_phi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/vllm_phi.py -------------------------------------------------------------------------------- /cosmos_curate/models/vllm_plugin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/vllm_plugin.py -------------------------------------------------------------------------------- /cosmos_curate/models/vllm_qwen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/models/vllm_qwen.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/README.md -------------------------------------------------------------------------------- /cosmos_curate/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/av/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/av/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/av/av_video_captioning_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/av/av_video_captioning_pipeline.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/av/av_video_ingesting_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/av/av_video_ingesting_pipeline.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/av/av_video_pipelines_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/av/av_video_pipelines_common.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/av/av_video_sharding_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/av/av_video_sharding_pipeline.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/av/av_video_splitting_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/av/av_video_splitting_pipeline.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/av/captioning/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/av/captioning/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/av/captioning/captioning_stages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/av/captioning/captioning_stages.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/av/clipping/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/av/clipping/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/av/clipping/clip_extraction_stages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/av/clipping/clip_extraction_stages.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/av/downloaders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/av/downloaders/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/av/downloaders/download_stages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/av/downloaders/download_stages.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/av/pipeline_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/av/pipeline_args.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/av/run_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/av/run_pipeline.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/av/trajectory/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/av/trajectory/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/av/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/av/utils/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/av/utils/av_data_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/av/utils/av_data_info.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/av/utils/av_data_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/av/utils/av_data_model.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/av/utils/av_pipe_input.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/av/utils/av_pipe_input.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/av/utils/postgres_schema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/av/utils/postgres_schema.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/av/utils/run_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/av/utils/run_utils.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/av/writers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/av/writers/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/av/writers/annotation_writer_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/av/writers/annotation_writer_stage.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/av/writers/base_writer_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/av/writers/base_writer_stage.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/av/writers/clip_writer_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/av/writers/clip_writer_stage.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/av/writers/cosmos_predict2_writer_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/av/writers/cosmos_predict2_writer_stage.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/av/writers/dataset_writer_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/av/writers/dataset_writer_stage.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/av/writers/make_db_row.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/av/writers/make_db_row.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/av/writers/t5_writer_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/av/writers/t5_writer_stage.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/av/writers/trajectory_writer_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/av/writers/trajectory_writer_stage.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/examples/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/examples/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/examples/demo_task_chunking_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/examples/demo_task_chunking_pipeline.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/examples/example_captioning_stages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/examples/example_captioning_stages.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/examples/hello_world_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/examples/hello_world_pipeline.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/pipeline_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/pipeline_args.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/captioning/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/captioning/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/captioning/api_caption_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/captioning/api_caption_stage.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/captioning/captioning_stages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/captioning/captioning_stages.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/captioning/vllm_caption_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/captioning/vllm_caption_stage.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/clipping/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/clipping/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/clipping/clip_extraction_stages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/clipping/clip_extraction_stages.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/clipping/clip_frame_extraction_stages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/clipping/clip_frame_extraction_stages.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/clipping/frame_extraction_stages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/clipping/frame_extraction_stages.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/clipping/transnetv2_extraction_stages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/clipping/transnetv2_extraction_stages.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/dedup/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/dedup/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/dedup/dedup_actor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/dedup/dedup_actor.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/dedup/raft_actor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/dedup/raft_actor.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/dedup_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/dedup_pipeline.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/embedding/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/embedding/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/embedding/cosmos_embed1_stages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/embedding/cosmos_embed1_stages.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/embedding/internvideo2_stages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/embedding/internvideo2_stages.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/filtering/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/filtering/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/filtering/aesthetics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/filtering/aesthetics/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/filtering/aesthetics/aesthetic_filter_stages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/filtering/aesthetics/aesthetic_filter_stages.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/filtering/aesthetics/qwen_filter_stages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/filtering/aesthetics/qwen_filter_stages.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/filtering/motion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/filtering/motion/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/filtering/motion/motion_filter_stages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/filtering/motion/motion_filter_stages.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/filtering/motion/motion_vector_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/filtering/motion/motion_vector_backend.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/preview/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/preview/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/preview/preview_stages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/preview/preview_stages.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/read_write/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/read_write/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/read_write/download_stages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/read_write/download_stages.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/read_write/metadata_writer_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/read_write/metadata_writer_stage.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/read_write/remux_stages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/read_write/remux_stages.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/read_write/summary_writers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/read_write/summary_writers.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/run_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/run_pipeline.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/sharding_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/sharding_pipeline.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/splitting_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/splitting_pipeline.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/utils/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/utils/data_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/utils/data_model.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/utils/decoder_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/utils/decoder_utils.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/utils/nvcodec_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/utils/nvcodec_utils.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/utils/video_pipe_input.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/utils/video_pipe_input.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/utils/vision_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/utils/vision_process.py -------------------------------------------------------------------------------- /cosmos_curate/pipelines/video/utils/windowing_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/pipelines/video/utils/windowing_utils.py -------------------------------------------------------------------------------- /cosmos_curate/scripts/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/scripts/__init__.py -------------------------------------------------------------------------------- /cosmos_curate/scripts/onto_nvcf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/scripts/onto_nvcf.py -------------------------------------------------------------------------------- /cosmos_curate/scripts/onto_slurm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/cosmos_curate/scripts/onto_slurm.py -------------------------------------------------------------------------------- /devset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/devset.sh -------------------------------------------------------------------------------- /docs/DEVELOPER_GUIDE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/docs/DEVELOPER_GUIDE.md -------------------------------------------------------------------------------- /docs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/docs/README.md -------------------------------------------------------------------------------- /docs/assets/cosmos-curate-dashboard.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/docs/assets/cosmos-curate-dashboard.png -------------------------------------------------------------------------------- /docs/assets/cosmos-curate-logical-view.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/docs/assets/cosmos-curate-logical-view.png -------------------------------------------------------------------------------- /docs/assets/cosmos-curate-physical-view.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/docs/assets/cosmos-curate-physical-view.png -------------------------------------------------------------------------------- /docs/assets/cosmos-curate-pipelines.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/docs/assets/cosmos-curate-pipelines.png -------------------------------------------------------------------------------- /docs/assets/nvidia-cosmos-header.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/docs/assets/nvidia-cosmos-header.png -------------------------------------------------------------------------------- /docs/client/END_USER_GUIDE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/docs/client/END_USER_GUIDE.md -------------------------------------------------------------------------------- /docs/client/NVCF_GUIDE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/docs/client/NVCF_GUIDE.md -------------------------------------------------------------------------------- /docs/curator/ARCHITECTURE_GUIDE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/docs/curator/ARCHITECTURE_GUIDE.md -------------------------------------------------------------------------------- /docs/curator/OBSERVABILITY_GUIDE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/docs/curator/OBSERVABILITY_GUIDE.md -------------------------------------------------------------------------------- /docs/curator/PIPELINE_DESIGN_GUIDE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/docs/curator/PIPELINE_DESIGN_GUIDE.md -------------------------------------------------------------------------------- /docs/curator/REFERENCE_PIPELINES_AV.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/docs/curator/REFERENCE_PIPELINES_AV.md -------------------------------------------------------------------------------- /docs/curator/REFERENCE_PIPELINES_VIDEO.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/docs/curator/REFERENCE_PIPELINES_VIDEO.md -------------------------------------------------------------------------------- /docs/curator/VLLM_INTERFACE_DEBUG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/docs/curator/VLLM_INTERFACE_DEBUG.md -------------------------------------------------------------------------------- /docs/curator/VLLM_INTERFACE_DESIGN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/docs/curator/VLLM_INTERFACE_DESIGN.md -------------------------------------------------------------------------------- /docs/curator/VLLM_INTERFACE_PLUGIN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/docs/curator/VLLM_INTERFACE_PLUGIN.md -------------------------------------------------------------------------------- /examples/nvcf/.gitignore: -------------------------------------------------------------------------------- 1 | filled_*.json 2 | -------------------------------------------------------------------------------- /examples/nvcf/ci/ci_create_helm_default.json.template: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/nvcf/ci/ci_create_helm_default.json.template -------------------------------------------------------------------------------- /examples/nvcf/ci/ci_deploy_helm_default.json.template: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/nvcf/ci/ci_deploy_helm_default.json.template -------------------------------------------------------------------------------- /examples/nvcf/creds/aws_credentials: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/nvcf/creds/aws_credentials -------------------------------------------------------------------------------- /examples/nvcf/creds/s3cred-function.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/nvcf/creds/s3cred-function.json -------------------------------------------------------------------------------- /examples/nvcf/function/create_curator_helm.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/nvcf/function/create_curator_helm.json -------------------------------------------------------------------------------- /examples/nvcf/function/deploy_curator_helm.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/nvcf/function/deploy_curator_helm.json -------------------------------------------------------------------------------- /examples/nvcf/function/invoke_video_dedup.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/nvcf/function/invoke_video_dedup.json -------------------------------------------------------------------------------- /examples/nvcf/function/invoke_video_shard.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/nvcf/function/invoke_video_shard.json -------------------------------------------------------------------------------- /examples/nvcf/function/invoke_video_split.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/nvcf/function/invoke_video_split.json -------------------------------------------------------------------------------- /examples/nvcf/function/invoke_video_split_full.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/nvcf/function/invoke_video_split_full.json -------------------------------------------------------------------------------- /examples/nvcf/image/image_upload.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/nvcf/image/image_upload.json -------------------------------------------------------------------------------- /examples/nvcf/model/model_upload_aesthetic.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/nvcf/model/model_upload_aesthetic.json -------------------------------------------------------------------------------- /examples/nvcf/model/model_upload_bert.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/nvcf/model/model_upload_bert.json -------------------------------------------------------------------------------- /examples/nvcf/model/model_upload_gemma_vllm.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/nvcf/model/model_upload_gemma_vllm.json -------------------------------------------------------------------------------- /examples/nvcf/model/model_upload_internvideo2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/nvcf/model/model_upload_internvideo2.json -------------------------------------------------------------------------------- /examples/nvcf/model/model_upload_kosmos2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/nvcf/model/model_upload_kosmos2.json -------------------------------------------------------------------------------- /examples/nvcf/model/model_upload_llava_mistral.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/nvcf/model/model_upload_llava_mistral.json -------------------------------------------------------------------------------- /examples/nvcf/model/model_upload_openai_clip.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/nvcf/model/model_upload_openai_clip.json -------------------------------------------------------------------------------- /examples/nvcf/model/model_upload_qwen_lm.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/nvcf/model/model_upload_qwen_lm.json -------------------------------------------------------------------------------- /examples/nvcf/model/model_upload_qwen_vl.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/nvcf/model/model_upload_qwen_vl.json -------------------------------------------------------------------------------- /examples/nvcf/model/model_upload_t5_xxl.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/nvcf/model/model_upload_t5_xxl.json -------------------------------------------------------------------------------- /examples/nvcf/model/model_upload_transnetv2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/nvcf/model/model_upload_transnetv2.json -------------------------------------------------------------------------------- /examples/observability/grafana/cosmos-curate-oss.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/observability/grafana/cosmos-curate-oss.json -------------------------------------------------------------------------------- /examples/slurm/ci_run_end_to_end.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/slurm/ci_run_end_to_end.sh -------------------------------------------------------------------------------- /examples/slurm/source_me_env_vars.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/slurm/source_me_env_vars.sh -------------------------------------------------------------------------------- /examples/slurm/source_me_source_code_mount.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/slurm/source_me_source_code_mount.sh -------------------------------------------------------------------------------- /examples/slurm/sync_config_creds_models.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/slurm/sync_config_creds_models.sh -------------------------------------------------------------------------------- /examples/slurm/sync_source_code.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/slurm/sync_source_code.sh -------------------------------------------------------------------------------- /examples/slurm/upload_image.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/slurm/upload_image.sh -------------------------------------------------------------------------------- /examples/workflow/.gitignore: -------------------------------------------------------------------------------- 1 | filled_*.json 2 | -------------------------------------------------------------------------------- /examples/workflow/example_input_file.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/workflow/example_input_file.json -------------------------------------------------------------------------------- /examples/workflow/example_nvcf_func_file.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/workflow/example_nvcf_func_file.json -------------------------------------------------------------------------------- /examples/workflow/operate_nvcf_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/workflow/operate_nvcf_functions.py -------------------------------------------------------------------------------- /examples/workflow/template_invoke_video_split.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/examples/workflow/template_invoke_video_split.json -------------------------------------------------------------------------------- /package/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/package/README.md -------------------------------------------------------------------------------- /package/cosmos_curate/default.dockerfile.jinja2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/package/cosmos_curate/default.dockerfile.jinja2 -------------------------------------------------------------------------------- /package/cosmos_curate/envs/paddle-ocr/post_install.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/package/cosmos_curate/envs/paddle-ocr/post_install.sh -------------------------------------------------------------------------------- /pixi.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/pixi.lock -------------------------------------------------------------------------------- /pixi.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/pixi.toml -------------------------------------------------------------------------------- /poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/poetry.lock -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/pyproject.toml -------------------------------------------------------------------------------- /pytest.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/pytest.ini -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/setup.py -------------------------------------------------------------------------------- /tests/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/README.md -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/__init__.py -------------------------------------------------------------------------------- /tests/benchmarks/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/benchmarks/conftest.py -------------------------------------------------------------------------------- /tests/benchmarks/split_pipeline/test_nvcf_split_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/benchmarks/split_pipeline/test_nvcf_split_benchmark.py -------------------------------------------------------------------------------- /tests/benchmarks/test_cloudevent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/benchmarks/test_cloudevent.py -------------------------------------------------------------------------------- /tests/benchmarks/test_secrets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/benchmarks/test_secrets.py -------------------------------------------------------------------------------- /tests/benchmarks/test_summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/benchmarks/test_summary.py -------------------------------------------------------------------------------- /tests/cosmos_curate/client/local_cli/test_launch_local.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/client/local_cli/test_launch_local.py -------------------------------------------------------------------------------- /tests/cosmos_curate/client/nvcf_cli/ncf/asset/test_asset_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/client/nvcf_cli/ncf/asset/test_asset_manager.py -------------------------------------------------------------------------------- /tests/cosmos_curate/client/nvcf_cli/ncf/common/test_nvcf_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/client/nvcf_cli/ncf/common/test_nvcf_base.py -------------------------------------------------------------------------------- /tests/cosmos_curate/client/nvcf_cli/ncf/common/test_nvcf_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/client/nvcf_cli/ncf/common/test_nvcf_client.py -------------------------------------------------------------------------------- /tests/cosmos_curate/client/nvcf_cli/ncf/config/test_config_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/client/nvcf_cli/ncf/config/test_config_manager.py -------------------------------------------------------------------------------- /tests/cosmos_curate/client/nvcf_cli/ncf/image/test_image_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/client/nvcf_cli/ncf/image/test_image_manager.py -------------------------------------------------------------------------------- /tests/cosmos_curate/client/nvcf_cli/ncf/launcher/test_nvcf_driver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/client/nvcf_cli/ncf/launcher/test_nvcf_driver.py -------------------------------------------------------------------------------- /tests/cosmos_curate/client/nvcf_cli/ncf/launcher/test_nvcf_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/client/nvcf_cli/ncf/launcher/test_nvcf_function.py -------------------------------------------------------------------------------- /tests/cosmos_curate/client/nvcf_cli/ncf/launcher/test_nvcf_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/client/nvcf_cli/ncf/launcher/test_nvcf_helper.py -------------------------------------------------------------------------------- /tests/cosmos_curate/client/nvcf_cli/ncf/model/test_model_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/client/nvcf_cli/ncf/model/test_model_manager.py -------------------------------------------------------------------------------- /tests/cosmos_curate/client/slurm_cli/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/client/slurm_cli/conftest.py -------------------------------------------------------------------------------- /tests/cosmos_curate/client/slurm_cli/test_prometheus_service_discovery.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/client/slurm_cli/test_prometheus_service_discovery.py -------------------------------------------------------------------------------- /tests/cosmos_curate/client/slurm_cli/test_slurm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/client/slurm_cli/test_slurm.py -------------------------------------------------------------------------------- /tests/cosmos_curate/client/utils/test_validations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/client/utils/test_validations.py -------------------------------------------------------------------------------- /tests/cosmos_curate/core/cf/test_nvcf_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/core/cf/test_nvcf_main.py -------------------------------------------------------------------------------- /tests/cosmos_curate/core/managers/test_model_cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/core/managers/test_model_cli.py -------------------------------------------------------------------------------- /tests/cosmos_curate/core/managers/test_postgres_cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/core/managers/test_postgres_cli.py -------------------------------------------------------------------------------- /tests/cosmos_curate/core/utils/dataset/test_dimensions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/core/utils/dataset/test_dimensions.py -------------------------------------------------------------------------------- /tests/cosmos_curate/core/utils/db/test_database_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/core/utils/db/test_database_utils.py -------------------------------------------------------------------------------- /tests/cosmos_curate/core/utils/misc/test_grouping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/core/utils/misc/test_grouping.py -------------------------------------------------------------------------------- /tests/cosmos_curate/core/utils/model/test_model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/core/utils/model/test_model_utils.py -------------------------------------------------------------------------------- /tests/cosmos_curate/core/utils/storage/test_presigned_s3_zip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/core/utils/storage/test_presigned_s3_zip.py -------------------------------------------------------------------------------- /tests/cosmos_curate/core/utils/storage/test_storage_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/core/utils/storage/test_storage_utils.py -------------------------------------------------------------------------------- /tests/cosmos_curate/core/utils/test_s3_prefix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/core/utils/test_s3_prefix.py -------------------------------------------------------------------------------- /tests/cosmos_curate/models/test_gpt2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/models/test_gpt2.py -------------------------------------------------------------------------------- /tests/cosmos_curate/models/test_prompts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/models/test_prompts.py -------------------------------------------------------------------------------- /tests/cosmos_curate/models/test_vllm_cosmos_reason1_vl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/models/test_vllm_cosmos_reason1_vl.py -------------------------------------------------------------------------------- /tests/cosmos_curate/models/test_vllm_interface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/models/test_vllm_interface.py -------------------------------------------------------------------------------- /tests/cosmos_curate/models/test_vllm_model_ids.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/models/test_vllm_model_ids.py -------------------------------------------------------------------------------- /tests/cosmos_curate/models/test_vllm_nemotron.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/models/test_vllm_nemotron.py -------------------------------------------------------------------------------- /tests/cosmos_curate/models/test_vllm_phi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/models/test_vllm_phi.py -------------------------------------------------------------------------------- /tests/cosmos_curate/models/test_vllm_qwen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/models/test_vllm_qwen.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/av/captioning/test_captioning_stages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/av/captioning/test_captioning_stages.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/av/clipping/test_clip_extraction_stages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/av/clipping/test_clip_extraction_stages.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/av/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/av/conftest.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/av/downloaders/test_download_stages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/av/downloaders/test_download_stages.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/av/utils/test_av_data_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/av/utils/test_av_data_model.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/av/utils/test_av_pipe_input.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/av/utils/test_av_pipe_input.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/av/writers/test_annotation_writer_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/av/writers/test_annotation_writer_stage.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/av/writers/test_clip_writer_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/av/writers/test_clip_writer_stage.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/av/writers/test_cosmos_predict2_writer_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/av/writers/test_cosmos_predict2_writer_stage.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/av/writers/test_dataset_writer_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/av/writers/test_dataset_writer_stage.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/av/writers/test_make_db_row.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/av/writers/test_make_db_row.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/av/writers/test_t5_writer_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/av/writers/test_t5_writer_stage.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/av/writers/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/av/writers/test_utils.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/conftest.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/video/captioning/example_t5_embeddings.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/video/captioning/example_t5_embeddings.pickle -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/video/captioning/test_api_caption_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/video/captioning/test_api_caption_stage.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/video/captioning/test_enhance_captions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/video/captioning/test_enhance_captions.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/video/captioning/test_qwen_captioning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/video/captioning/test_qwen_captioning.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/video/captioning/test_t5_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/video/captioning/test_t5_embedding.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/video/captioning/test_vllm_caption_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/video/captioning/test_vllm_caption_stage.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/video/clipping/test_fixed_stride_extraction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/video/clipping/test_fixed_stride_extraction.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/video/clipping/test_transnetv2_extraction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/video/clipping/test_transnetv2_extraction.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/video/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/video/conftest.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/video/embedding/test_cosmos_embed1_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/video/embedding/test_cosmos_embed1_embedding.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/video/embedding/test_internvideo2_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/video/embedding/test_internvideo2_embedding.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/video/filtering/aesthetics/test_aesthetic_filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/video/filtering/aesthetics/test_aesthetic_filter.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/video/filtering/aesthetics/test_qwen_filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/video/filtering/aesthetics/test_qwen_filter.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/video/filtering/motion/test_motion_filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/video/filtering/motion/test_motion_filter.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/video/read_write/test_metadata_writer_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/video/read_write/test_metadata_writer_stage.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/video/read_write/test_remux_stages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/video/read_write/test_remux_stages.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/video/utils/test_data_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/video/utils/test_data_model.py -------------------------------------------------------------------------------- /tests/cosmos_curate/pipelines/video/utils/test_decoder_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/pipelines/video/utils/test_decoder_utils.py -------------------------------------------------------------------------------- /tests/cosmos_curate/scripts/test_onto_slurm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/cosmos_curate/scripts/test_onto_slurm.py -------------------------------------------------------------------------------- /tests/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/utils/__init__.py -------------------------------------------------------------------------------- /tests/utils/gpu_worker_limiter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/utils/gpu_worker_limiter.py -------------------------------------------------------------------------------- /tests/utils/sequential_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/utils/sequential_runner.py -------------------------------------------------------------------------------- /tests/utils/vllm_mock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nvidia-cosmos/cosmos-curate/HEAD/tests/utils/vllm_mock.py --------------------------------------------------------------------------------