├── .all-contributorsrc ├── .envrc ├── .git_archival.txt ├── .gitattributes ├── .github ├── dependabot.yml └── workflows │ ├── ci.yaml │ ├── codeql.yml │ ├── docs.yaml │ ├── llm-docs.yml │ └── pypi-package.yaml ├── .gitignore ├── .golangci.yaml ├── .goreleaser.yaml ├── .mockery.yml ├── .python-version ├── .tool-versions ├── .vscode ├── extensions.json └── settings.json ├── CLAUDE.md ├── CONTRIBUTING.md ├── LICENSE ├── Makefile ├── README.md ├── cmd ├── base-image │ └── baseimage.go └── cog │ └── cog.go ├── docs ├── CNAME ├── cli.md ├── cog.png ├── deploy.md ├── environment.md ├── getting-started-own-model.md ├── getting-started.md ├── http.md ├── llms.txt ├── notebooks.md ├── private-package-registry.md ├── python.md ├── redis.md ├── stylesheets │ └── extra.css ├── training.md ├── wsl2 │ ├── images │ │ ├── cog_model_output.png │ │ ├── enable_feature_success.png │ │ ├── glide_out.png │ │ ├── memory-usage.png │ │ ├── nvidia_driver_select.png │ │ └── wsl2-enable.png │ └── wsl2.md └── yaml.md ├── go.mod ├── go.sum ├── mkdocs.yml ├── pkg ├── api │ ├── client.go │ └── client_test.go ├── cli │ ├── baseimage.go │ ├── build.go │ ├── debug.go │ ├── init-templates │ │ ├── base │ │ │ ├── .dockerignore │ │ │ ├── .github │ │ │ │ └── workflows │ │ │ │ │ └── push.yaml │ │ │ ├── cog.yaml │ │ │ ├── predict.py │ │ │ └── requirements.txt │ │ └── pipeline │ │ │ ├── .dockerignore │ │ │ ├── AGENTS.md │ │ │ ├── README.md │ │ │ ├── cog.yaml │ │ │ ├── main.py │ │ │ └── requirements.txt │ ├── init.go │ ├── init_test.go │ ├── login.go │ ├── migrate.go │ ├── predict.go │ ├── pull.go │ ├── push.go │ ├── root.go │ ├── run.go │ ├── serve.go │ └── train.go ├── coglog │ ├── build_log_context.go │ ├── client.go │ ├── client_test.go │ ├── env.go │ ├── env_test.go │ ├── migrate_log_context.go │ ├── pull_log_context.go │ └── push_log_context.go ├── config │ ├── compatibility.go │ ├── compatibility_test.go │ ├── config.go │ ├── config_test.go │ ├── cuda_base_images.json │ ├── data │ │ └── config_schema_v1.0.json │ ├── env_variables.go │ ├── env_variables_test.go │ ├── image_name.go │ ├── image_name_test.go │ ├── load.go │ ├── load_test.go │ ├── tf_compatibility_matrix.json │ ├── torch_compatibility_matrix.json │ ├── validator.go │ ├── validator_test.go │ └── version.go ├── docker │ ├── api_client.go │ ├── build_secrets.go │ ├── buildkit.go │ ├── command │ │ ├── command.go │ │ ├── errors.go │ │ ├── manifest.go │ │ └── user_info.go │ ├── credential_helper_input.go │ ├── credentials.go │ ├── credentials_test.go │ ├── docker.go │ ├── docker_client_test.go │ ├── docker_command.go │ ├── docker_command_test.go │ ├── dockertest │ │ ├── command_mocks.go │ │ ├── helper_client.go │ │ ├── image.go │ │ ├── mock_command.go │ │ ├── ref.go │ │ ├── ref_test.go │ │ └── testdata │ │ │ ├── alpine.tar │ │ │ └── create-image-fixtures.sh │ ├── env.go │ ├── errors.go │ ├── fast_push.go │ ├── fast_push_test.go │ ├── host.go │ ├── host_unix.go │ ├── host_windows.go │ ├── login.go │ ├── monobase.go │ ├── monobase_test.go │ ├── options.go │ ├── pipeline_push.go │ ├── pipeline_push_test.go │ ├── push.go │ ├── push_test.go │ ├── run.go │ ├── run_test.go │ ├── standard_push.go │ └── standard_push_test.go ├── dockercontext │ ├── build_tempdir.go │ ├── build_tempdir_test.go │ └── directories.go ├── dockerfile │ ├── base.go │ ├── base_test.go │ ├── cog_embed.go │ ├── cog_embed_test.go │ ├── env.go │ ├── fast_generator.go │ ├── fast_generator_test.go │ ├── generator.go │ ├── generator_factory.go │ ├── generator_factory_test.go │ ├── monobase_matrix.go │ ├── monobase_matrix_test.go │ ├── standard_generator.go │ ├── standard_generator_test.go │ └── version_check.go ├── dockerignore │ ├── dockerignore.go │ └── dockerignore_test.go ├── env │ ├── env.go │ └── env_test.go ├── errors │ ├── common.go │ └── errors.go ├── global │ └── global.go ├── http │ ├── client.go │ ├── client_test.go │ ├── transport.go │ ├── transport_test.go │ ├── user_agent.go │ └── user_agent_test.go ├── image │ ├── build.go │ ├── build_test.go │ ├── config.go │ ├── model_dependencies.go │ ├── openapi_schema.go │ └── pip_freeze.go ├── migrate │ ├── factory.go │ ├── factory_test.go │ ├── migrations.go │ ├── migrations_test.go │ ├── migrator.go │ ├── migrator_v1_v1fast.go │ └── migrator_v1_v1fast_test.go ├── monobeam │ ├── client.go │ └── client_test.go ├── path │ ├── path.go │ └── path_test.go ├── predict │ ├── api.go │ ├── input.go │ └── predictor.go ├── procedure │ ├── validate.go │ └── validate_test.go ├── registry │ ├── client.go │ ├── client_test.go │ ├── manifest_result.go │ ├── registry_client.go │ └── registrytest │ │ └── mock_client.go ├── registry_testhelpers │ ├── registry_container.go │ └── testdata │ │ └── docker │ │ └── registry │ │ └── v2 │ │ ├── blobs │ │ └── sha256 │ │ │ ├── 75 │ │ │ └── 757d680068d77be46fd1ea20fb21db16f150468c5e7079a08a2e4705aec096ac │ │ │ │ └── data │ │ │ ├── 1c │ │ │ └── 1c4eef651f65e2f7daee7ee785882ac164b02b78fb74503052a26dc061c90474 │ │ │ │ └── data │ │ │ ├── 6e │ │ │ └── 6e771e15690e2fabf2332d3a3b744495411d6e0b00b2aea64419b58b0066cf81 │ │ │ │ └── data │ │ │ ├── 8d │ │ │ └── 8d591b0b7dea080ea3be9e12ae563eebf9869168ffced1cb25b2470a3d9fe15e │ │ │ │ └── data │ │ │ ├── 9a │ │ │ └── 9a0ff41dccad7a96f324a4655a715c623ed3511c7336361ffa9dadcecbdb99e5 │ │ │ │ └── data │ │ │ ├── ad │ │ │ └── aded1e1a5b3705116fa0a92ba074a5e0b0031647d9c315983ccba2ee5428ec8b │ │ │ │ └── data │ │ │ └── f1 │ │ │ └── f18232174bc91741fdf3da96d85011092101a032a93a388b79e99e69c2d5c870 │ │ │ └── data │ │ └── repositories │ │ └── alpine │ │ ├── _layers │ │ └── sha256 │ │ │ ├── 6e771e15690e2fabf2332d3a3b744495411d6e0b00b2aea64419b58b0066cf81 │ │ │ └── link │ │ │ ├── 8d591b0b7dea080ea3be9e12ae563eebf9869168ffced1cb25b2470a3d9fe15e │ │ │ └── link │ │ │ ├── aded1e1a5b3705116fa0a92ba074a5e0b0031647d9c315983ccba2ee5428ec8b │ │ │ └── link │ │ │ └── f18232174bc91741fdf3da96d85011092101a032a93a388b79e99e69c2d5c870 │ │ │ └── link │ │ └── _manifests │ │ ├── revisions │ │ └── sha256 │ │ │ ├── 1c4eef651f65e2f7daee7ee785882ac164b02b78fb74503052a26dc061c90474 │ │ │ └── link │ │ │ ├── 757d680068d77be46fd1ea20fb21db16f150468c5e7079a08a2e4705aec096ac │ │ │ └── link │ │ │ └── 9a0ff41dccad7a96f324a4655a715c623ed3511c7336361ffa9dadcecbdb99e5 │ │ │ └── link │ │ └── tags │ │ └── latest │ │ ├── current │ │ └── link │ │ └── index │ │ └── sha256 │ │ └── 9a0ff41dccad7a96f324a4655a715c623ed3511c7336361ffa9dadcecbdb99e5 │ │ └── link ├── requirements │ ├── requirements.go │ └── requirements_test.go ├── update │ ├── state.go │ └── update.go ├── util │ ├── console │ │ ├── console.go │ │ ├── formatting.go │ │ ├── global.go │ │ ├── interactive.go │ │ ├── levels.go │ │ └── term.go │ ├── env.go │ ├── errors.go │ ├── files │ │ ├── files.go │ │ └── files_test.go │ ├── hash.go │ ├── hash_test.go │ ├── mime │ │ ├── mime.go │ │ └── mime_test.go │ ├── net.go │ ├── overwrite_yaml.go │ ├── overwrite_yaml_test.go │ ├── platform.go │ ├── ringbuffer.go │ ├── shell │ │ ├── net.go │ │ └── pipes.go │ ├── slices │ │ └── slices.go │ └── version │ │ ├── version.go │ │ └── version_test.go ├── web │ ├── client.go │ └── client_test.go └── weights │ ├── fast_weights.go │ ├── fast_weights_manifest.go │ ├── fast_weights_test.go │ ├── manifest.go │ ├── weights.go │ └── weights_test.go ├── pyproject.toml ├── python ├── cog │ ├── .gitignore │ ├── __init__.py │ ├── base_input.py │ ├── base_predictor.py │ ├── code_xforms.py │ ├── command │ │ ├── __init__.py │ │ ├── ast_openapi_schema.py │ │ ├── call_graph.py │ │ ├── migrate_v1_v1fast.py │ │ └── openapi_schema.py │ ├── config.py │ ├── env_property.py │ ├── errors.py │ ├── ext │ │ └── __init__.py │ ├── files.py │ ├── json.py │ ├── logging.py │ ├── mimetypes_ext.py │ ├── mode.py │ ├── predictor.py │ ├── schema.py │ ├── server │ │ ├── __init__.py │ │ ├── connection.py │ │ ├── errors.py │ │ ├── eventtypes.py │ │ ├── exceptions.py │ │ ├── helpers.py │ │ ├── http.py │ │ ├── probes.py │ │ ├── response_throttler.py │ │ ├── runner.py │ │ ├── scope.py │ │ ├── telemetry.py │ │ ├── useragent.py │ │ ├── webhook.py │ │ └── worker.py │ ├── suppress_output.py │ ├── types.py │ └── wait.py └── tests │ ├── cog │ └── test_files.py │ ├── command │ ├── __init__.py │ └── call_graph_test.py │ ├── conftest.py │ ├── fixtures │ └── argv_override.py │ ├── server │ ├── __init__.py │ ├── conftest.py │ ├── fixtures │ │ ├── async_setup_uses_same_loop_as_predict.py │ │ ├── catch_in_predict.py │ │ ├── complex_output.py │ │ ├── count_up.py │ │ ├── exc_in_predict.py │ │ ├── exc_in_setup.py │ │ ├── exc_in_setup_and_predict.py │ │ ├── exc_on_import.py │ │ ├── exit_in_predict.py │ │ ├── exit_in_setup.py │ │ ├── exit_on_import.py │ │ ├── function.py │ │ ├── hello_world.py │ │ ├── hello_world_async.py │ │ ├── import_err.py │ │ ├── input_choices.py │ │ ├── input_choices_integer.py │ │ ├── input_choices_iterable.py │ │ ├── input_deprecated.py │ │ ├── input_file.py │ │ ├── input_ge_le.py │ │ ├── input_integer.py │ │ ├── input_integer_default.py │ │ ├── input_kwargs.py │ │ ├── input_literal.py │ │ ├── input_literal_integer.py │ │ ├── input_multiple.py │ │ ├── input_none.py │ │ ├── input_path.py │ │ ├── input_path_2.py │ │ ├── input_path_or_none.py │ │ ├── input_secret.py │ │ ├── input_string.py │ │ ├── input_union_integer_or_list_of_integers.py │ │ ├── input_union_string_or_list_of_strings.py │ │ ├── input_unsupported_type.py │ │ ├── input_untyped.py │ │ ├── killed_in_predict.py │ │ ├── logging.py │ │ ├── logging_async.py │ │ ├── missing_predict.py │ │ ├── missing_predictor.py │ │ ├── openapi_complex_input.py │ │ ├── openapi_custom_output_type.py │ │ ├── openapi_input_int_choices.py │ │ ├── openapi_optional_output_type.py │ │ ├── openapi_output_list.py │ │ ├── openapi_output_type.py │ │ ├── openapi_output_yield.py │ │ ├── output_complex.py │ │ ├── output_file.py │ │ ├── output_file_named.py │ │ ├── output_iterator_complex.py │ │ ├── output_numpy.py │ │ ├── output_path_image.py │ │ ├── output_path_text.py │ │ ├── output_wrong_type.py │ │ ├── record_metric.py │ │ ├── record_metric_async.py │ │ ├── setup.py │ │ ├── setup_async.py │ │ ├── setup_async_with_sync_predict.py │ │ ├── setup_uses_async.py │ │ ├── setup_weights.py │ │ ├── simple.py │ │ ├── simple_async.py │ │ ├── sleep.py │ │ ├── sleep_async.py │ │ ├── slow_predict.py │ │ ├── slow_setup.py │ │ ├── steps.py │ │ ├── stream_redirector_race_condition.py │ │ ├── train.py │ │ ├── with_context.py │ │ ├── with_context_async.py │ │ ├── yield_concatenate_iterator.py │ │ ├── yield_files.py │ │ ├── yield_strings.py │ │ └── yield_strings_file_input.py │ ├── test_helpers.py │ ├── test_http.py │ ├── test_http_input.py │ ├── test_http_output.py │ ├── test_predictor.py │ ├── test_probes.py │ ├── test_response_throttler.py │ ├── test_runner.py │ ├── test_webhook.py │ └── test_worker.py │ ├── test_code_xforms.py │ ├── test_config.py │ ├── test_json.py │ ├── test_mimetypes_ext.py │ ├── test_predictor.py │ ├── test_types.py │ └── test_wait.py ├── script ├── format ├── lint ├── setup ├── test ├── test-all ├── test-go └── test-python ├── test-integration └── test_integration │ ├── __init__.py │ ├── conftest.py │ ├── fixtures │ ├── apt-packages │ │ ├── cog.yaml │ │ └── predict.py │ ├── async-sleep-project │ │ ├── cog.yaml │ │ └── predict.py │ ├── async-string-project │ │ ├── cog.yaml │ │ └── predict.py │ ├── bad-dockerignore │ │ ├── .dockerignore │ │ ├── cog.yaml │ │ └── predict.py │ ├── cog-runtime-float │ │ ├── cog.yaml │ │ └── predict.py │ ├── cog-runtime-int │ │ ├── cog.yaml │ │ └── predict.py │ ├── complex-types-list-project │ │ ├── cog.yaml │ │ ├── predict.py │ │ └── requirements.txt │ ├── complex-types │ │ ├── cog.yaml │ │ ├── predict.py │ │ └── requirements.txt │ ├── complex_output_project │ │ ├── cog.yaml │ │ └── predict.py │ ├── env-project │ │ ├── cog.yaml │ │ └── predict.py │ ├── fast-build │ │ ├── cog.yaml │ │ ├── predict.py │ │ └── requirements.txt │ ├── ffmpeg-package │ │ ├── cog.yaml │ │ └── predict.py │ ├── file-input-project │ │ ├── cog.yaml │ │ ├── predict.py │ │ └── test.txt │ ├── file-list-input-project │ │ ├── cog.yaml │ │ └── predict.py │ ├── future-annotations-project │ │ ├── cog.yaml │ │ ├── predict.py │ │ └── some_image.jpg │ ├── glb-project │ │ ├── cog.yaml │ │ ├── mesh.glb │ │ ├── output.txt │ │ └── predict.py │ ├── granite-project │ │ ├── cog.yaml │ │ ├── predict.py │ │ └── requirements.txt │ ├── install-requires-packaging │ │ ├── cog.yaml │ │ ├── predict.py │ │ └── requirements.txt │ ├── int-none-output-project │ │ ├── cog.yaml │ │ └── predict.py │ ├── int-project │ │ ├── cog.yaml │ │ └── predict.py │ ├── invalid-int-project │ │ ├── cog.yaml │ │ └── predict.py │ ├── local-whl-install │ │ ├── cog.yaml │ │ ├── my_package-0.1-py2.py3-none-any.whl │ │ ├── predict.py │ │ └── requirements.txt │ ├── many-inputs-project │ │ ├── cog.yaml │ │ └── predict.py │ ├── migration-gpu-project │ │ ├── cog.yaml │ │ └── predict.py │ ├── migration-no-python-changes-project │ │ ├── cog.yaml │ │ └── predict.py │ ├── migration-project │ │ ├── cog.yaml │ │ └── predict.py │ ├── new-union-project │ │ ├── cog.yaml │ │ └── predict.py │ ├── no-predictor-project │ │ └── cog.yaml │ ├── optional-project │ │ ├── cog.yaml │ │ └── predict.py │ ├── overrides-project │ │ ├── cog.yaml │ │ ├── overrides.txt │ │ ├── predict.py │ │ └── requirements.txt │ ├── path-input-project │ │ ├── cog.yaml │ │ └── predict.py │ ├── path-list-input-project │ │ ├── cog.yaml │ │ └── predict.py │ ├── path-list-output-project │ │ ├── cog.yaml │ │ └── predict.py │ ├── path-output-project │ │ ├── cog.yaml │ │ └── predict.py │ ├── path-project │ │ ├── cog.yaml │ │ └── predict.py │ ├── pipeline-project │ │ ├── cog.yaml │ │ └── predict.py │ ├── pipeline-requirements-project │ │ ├── .gitignore │ │ ├── cog.yaml │ │ └── predict.py │ ├── procedure-project │ │ ├── cog.yaml │ │ └── predict.py │ ├── pydantic1-none │ │ ├── cog.yaml │ │ └── predict.py │ ├── pydantic2-output │ │ ├── cog.yaml │ │ ├── predict.py │ │ └── train.py │ ├── pydantic2 │ │ ├── cog.yaml │ │ └── predict.py │ ├── python-313 │ │ ├── cog.yaml │ │ └── predict.py │ ├── python_37 │ │ ├── cog.yaml │ │ └── predict.py │ ├── secrets-project │ │ ├── cog.yaml │ │ ├── file-secret.txt │ │ └── predict.py │ ├── setup-subprocess-double-fork-http-project │ │ ├── cog.yaml │ │ ├── pong.py │ │ ├── predict.py │ │ └── run-pong.sh │ ├── setup-subprocess-double-fork-project │ │ ├── .gitignore │ │ ├── cog.yaml │ │ ├── forker.py │ │ ├── predict.py │ │ └── run-forker.sh │ ├── setup-subprocess-multiprocessing-project │ │ ├── .gitignore │ │ ├── bg.py │ │ ├── cog.yaml │ │ └── predict.py │ ├── setup-subprocess-simple-project │ │ ├── child.sh │ │ ├── cog.yaml │ │ └── predict.py │ ├── string-list-project │ │ ├── cog.yaml │ │ └── predict.py │ ├── string-none-output-project │ │ ├── cog.yaml │ │ └── predict.py │ ├── string-project │ │ ├── cog.yaml │ │ ├── input.json │ │ └── predict.py │ ├── subdirectory-project │ │ ├── cog.yaml │ │ ├── my-subdir │ │ │ └── predict.py │ │ └── mylib.py │ ├── tensorflow-project │ │ ├── .requirements.txt │ │ ├── cog.yaml │ │ └── predict.py │ ├── torch-270-cuda-126 │ │ ├── cog.yaml │ │ └── predict.py │ ├── torch-271-cuda-128 │ │ ├── cog.yaml │ │ └── predict.py │ ├── torch-baseimage-project │ │ ├── cog.yaml │ │ ├── openapi.json │ │ └── predict.py │ ├── torch-cuda-baseimage-project │ │ ├── cog.yaml │ │ └── predict.py │ ├── train-project │ │ ├── cog.yaml │ │ ├── predict.py │ │ └── train.py │ ├── training-setup-project │ │ ├── cog.yaml │ │ ├── predict.py │ │ └── train.py │ └── zsh-package │ │ ├── cog.yaml │ │ ├── predict.py │ │ └── requirements.txt │ ├── test_build.py │ ├── test_config.py │ ├── test_migrate.py │ ├── test_predict.py │ ├── test_run.py │ ├── test_train.py │ └── util.py ├── tools ├── compatgen │ ├── internal │ │ ├── cuda.go │ │ ├── tensorflow.go │ │ ├── torch.go │ │ ├── torch_package.go │ │ ├── torch_test.go │ │ ├── torch_test.html │ │ └── util.go │ └── main.go ├── install.sh ├── test-registry-util │ ├── README.md │ └── main.go └── uploader │ ├── iface.go │ └── s3.go ├── tox.ini └── uv.lock /.all-contributorsrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/.all-contributorsrc -------------------------------------------------------------------------------- /.envrc: -------------------------------------------------------------------------------- 1 | use asdf 2 | layout python 3 | dotenv_if_exists 4 | -------------------------------------------------------------------------------- /.git_archival.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/.git_archival.txt -------------------------------------------------------------------------------- /.gitattributes: -------------------------------------------------------------------------------- 1 | .git_archival.txt export-subst 2 | Makefile -linguist-detectable 3 | -------------------------------------------------------------------------------- /.github/dependabot.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/.github/dependabot.yml -------------------------------------------------------------------------------- /.github/workflows/ci.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/.github/workflows/ci.yaml -------------------------------------------------------------------------------- /.github/workflows/codeql.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/.github/workflows/codeql.yml -------------------------------------------------------------------------------- /.github/workflows/docs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/.github/workflows/docs.yaml -------------------------------------------------------------------------------- /.github/workflows/llm-docs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/.github/workflows/llm-docs.yml -------------------------------------------------------------------------------- /.github/workflows/pypi-package.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/.github/workflows/pypi-package.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/.gitignore -------------------------------------------------------------------------------- /.golangci.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/.golangci.yaml -------------------------------------------------------------------------------- /.goreleaser.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/.goreleaser.yaml -------------------------------------------------------------------------------- /.mockery.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/.mockery.yml -------------------------------------------------------------------------------- /.python-version: -------------------------------------------------------------------------------- 1 | 3.13.2 2 | -------------------------------------------------------------------------------- /.tool-versions: -------------------------------------------------------------------------------- 1 | golang 1.24.2 2 | python 3.13.2 3 | -------------------------------------------------------------------------------- /.vscode/extensions.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/.vscode/extensions.json -------------------------------------------------------------------------------- /.vscode/settings.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/.vscode/settings.json -------------------------------------------------------------------------------- /CLAUDE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/CLAUDE.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/README.md -------------------------------------------------------------------------------- /cmd/base-image/baseimage.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/cmd/base-image/baseimage.go -------------------------------------------------------------------------------- /cmd/cog/cog.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/cmd/cog/cog.go -------------------------------------------------------------------------------- /docs/CNAME: -------------------------------------------------------------------------------- 1 | cog.run 2 | -------------------------------------------------------------------------------- /docs/cli.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/docs/cli.md -------------------------------------------------------------------------------- /docs/cog.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/docs/cog.png -------------------------------------------------------------------------------- /docs/deploy.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/docs/deploy.md -------------------------------------------------------------------------------- /docs/environment.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/docs/environment.md -------------------------------------------------------------------------------- /docs/getting-started-own-model.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/docs/getting-started-own-model.md -------------------------------------------------------------------------------- /docs/getting-started.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/docs/getting-started.md -------------------------------------------------------------------------------- /docs/http.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/docs/http.md -------------------------------------------------------------------------------- /docs/llms.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/docs/llms.txt -------------------------------------------------------------------------------- /docs/notebooks.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/docs/notebooks.md -------------------------------------------------------------------------------- /docs/private-package-registry.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/docs/private-package-registry.md -------------------------------------------------------------------------------- /docs/python.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/docs/python.md -------------------------------------------------------------------------------- /docs/redis.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/docs/redis.md -------------------------------------------------------------------------------- /docs/stylesheets/extra.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/docs/stylesheets/extra.css -------------------------------------------------------------------------------- /docs/training.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/docs/training.md -------------------------------------------------------------------------------- /docs/wsl2/images/cog_model_output.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/docs/wsl2/images/cog_model_output.png -------------------------------------------------------------------------------- /docs/wsl2/images/enable_feature_success.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/docs/wsl2/images/enable_feature_success.png -------------------------------------------------------------------------------- /docs/wsl2/images/glide_out.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/docs/wsl2/images/glide_out.png -------------------------------------------------------------------------------- /docs/wsl2/images/memory-usage.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/docs/wsl2/images/memory-usage.png -------------------------------------------------------------------------------- /docs/wsl2/images/nvidia_driver_select.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/docs/wsl2/images/nvidia_driver_select.png -------------------------------------------------------------------------------- /docs/wsl2/images/wsl2-enable.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/docs/wsl2/images/wsl2-enable.png -------------------------------------------------------------------------------- /docs/wsl2/wsl2.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/docs/wsl2/wsl2.md -------------------------------------------------------------------------------- /docs/yaml.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/docs/yaml.md -------------------------------------------------------------------------------- /go.mod: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/go.mod -------------------------------------------------------------------------------- /go.sum: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/go.sum -------------------------------------------------------------------------------- /mkdocs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/mkdocs.yml -------------------------------------------------------------------------------- /pkg/api/client.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/api/client.go -------------------------------------------------------------------------------- /pkg/api/client_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/api/client_test.go -------------------------------------------------------------------------------- /pkg/cli/baseimage.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/cli/baseimage.go -------------------------------------------------------------------------------- /pkg/cli/build.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/cli/build.go -------------------------------------------------------------------------------- /pkg/cli/debug.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/cli/debug.go -------------------------------------------------------------------------------- /pkg/cli/init-templates/base/.dockerignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/cli/init-templates/base/.dockerignore -------------------------------------------------------------------------------- /pkg/cli/init-templates/base/.github/workflows/push.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/cli/init-templates/base/.github/workflows/push.yaml -------------------------------------------------------------------------------- /pkg/cli/init-templates/base/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/cli/init-templates/base/cog.yaml -------------------------------------------------------------------------------- /pkg/cli/init-templates/base/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/cli/init-templates/base/predict.py -------------------------------------------------------------------------------- /pkg/cli/init-templates/base/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/cli/init-templates/base/requirements.txt -------------------------------------------------------------------------------- /pkg/cli/init-templates/pipeline/.dockerignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/cli/init-templates/pipeline/.dockerignore -------------------------------------------------------------------------------- /pkg/cli/init-templates/pipeline/AGENTS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/cli/init-templates/pipeline/AGENTS.md -------------------------------------------------------------------------------- /pkg/cli/init-templates/pipeline/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/cli/init-templates/pipeline/README.md -------------------------------------------------------------------------------- /pkg/cli/init-templates/pipeline/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/cli/init-templates/pipeline/cog.yaml -------------------------------------------------------------------------------- /pkg/cli/init-templates/pipeline/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/cli/init-templates/pipeline/main.py -------------------------------------------------------------------------------- /pkg/cli/init-templates/pipeline/requirements.txt: -------------------------------------------------------------------------------- 1 | replicate>=1.1.0b1,<2.0.0a1 2 | -------------------------------------------------------------------------------- /pkg/cli/init.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/cli/init.go -------------------------------------------------------------------------------- /pkg/cli/init_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/cli/init_test.go -------------------------------------------------------------------------------- /pkg/cli/login.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/cli/login.go -------------------------------------------------------------------------------- /pkg/cli/migrate.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/cli/migrate.go -------------------------------------------------------------------------------- /pkg/cli/predict.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/cli/predict.go -------------------------------------------------------------------------------- /pkg/cli/pull.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/cli/pull.go -------------------------------------------------------------------------------- /pkg/cli/push.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/cli/push.go -------------------------------------------------------------------------------- /pkg/cli/root.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/cli/root.go -------------------------------------------------------------------------------- /pkg/cli/run.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/cli/run.go -------------------------------------------------------------------------------- /pkg/cli/serve.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/cli/serve.go -------------------------------------------------------------------------------- /pkg/cli/train.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/cli/train.go -------------------------------------------------------------------------------- /pkg/coglog/build_log_context.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/coglog/build_log_context.go -------------------------------------------------------------------------------- /pkg/coglog/client.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/coglog/client.go -------------------------------------------------------------------------------- /pkg/coglog/client_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/coglog/client_test.go -------------------------------------------------------------------------------- /pkg/coglog/env.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/coglog/env.go -------------------------------------------------------------------------------- /pkg/coglog/env_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/coglog/env_test.go -------------------------------------------------------------------------------- /pkg/coglog/migrate_log_context.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/coglog/migrate_log_context.go -------------------------------------------------------------------------------- /pkg/coglog/pull_log_context.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/coglog/pull_log_context.go -------------------------------------------------------------------------------- /pkg/coglog/push_log_context.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/coglog/push_log_context.go -------------------------------------------------------------------------------- /pkg/config/compatibility.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/config/compatibility.go -------------------------------------------------------------------------------- /pkg/config/compatibility_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/config/compatibility_test.go -------------------------------------------------------------------------------- /pkg/config/config.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/config/config.go -------------------------------------------------------------------------------- /pkg/config/config_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/config/config_test.go -------------------------------------------------------------------------------- /pkg/config/cuda_base_images.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/config/cuda_base_images.json -------------------------------------------------------------------------------- /pkg/config/data/config_schema_v1.0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/config/data/config_schema_v1.0.json -------------------------------------------------------------------------------- /pkg/config/env_variables.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/config/env_variables.go -------------------------------------------------------------------------------- /pkg/config/env_variables_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/config/env_variables_test.go -------------------------------------------------------------------------------- /pkg/config/image_name.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/config/image_name.go -------------------------------------------------------------------------------- /pkg/config/image_name_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/config/image_name_test.go -------------------------------------------------------------------------------- /pkg/config/load.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/config/load.go -------------------------------------------------------------------------------- /pkg/config/load_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/config/load_test.go -------------------------------------------------------------------------------- /pkg/config/tf_compatibility_matrix.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/config/tf_compatibility_matrix.json -------------------------------------------------------------------------------- /pkg/config/torch_compatibility_matrix.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/config/torch_compatibility_matrix.json -------------------------------------------------------------------------------- /pkg/config/validator.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/config/validator.go -------------------------------------------------------------------------------- /pkg/config/validator_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/config/validator_test.go -------------------------------------------------------------------------------- /pkg/config/version.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/config/version.go -------------------------------------------------------------------------------- /pkg/docker/api_client.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/api_client.go -------------------------------------------------------------------------------- /pkg/docker/build_secrets.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/build_secrets.go -------------------------------------------------------------------------------- /pkg/docker/buildkit.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/buildkit.go -------------------------------------------------------------------------------- /pkg/docker/command/command.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/command/command.go -------------------------------------------------------------------------------- /pkg/docker/command/errors.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/command/errors.go -------------------------------------------------------------------------------- /pkg/docker/command/manifest.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/command/manifest.go -------------------------------------------------------------------------------- /pkg/docker/command/user_info.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/command/user_info.go -------------------------------------------------------------------------------- /pkg/docker/credential_helper_input.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/credential_helper_input.go -------------------------------------------------------------------------------- /pkg/docker/credentials.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/credentials.go -------------------------------------------------------------------------------- /pkg/docker/credentials_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/credentials_test.go -------------------------------------------------------------------------------- /pkg/docker/docker.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/docker.go -------------------------------------------------------------------------------- /pkg/docker/docker_client_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/docker_client_test.go -------------------------------------------------------------------------------- /pkg/docker/docker_command.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/docker_command.go -------------------------------------------------------------------------------- /pkg/docker/docker_command_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/docker_command_test.go -------------------------------------------------------------------------------- /pkg/docker/dockertest/command_mocks.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/dockertest/command_mocks.go -------------------------------------------------------------------------------- /pkg/docker/dockertest/helper_client.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/dockertest/helper_client.go -------------------------------------------------------------------------------- /pkg/docker/dockertest/image.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/dockertest/image.go -------------------------------------------------------------------------------- /pkg/docker/dockertest/mock_command.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/dockertest/mock_command.go -------------------------------------------------------------------------------- /pkg/docker/dockertest/ref.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/dockertest/ref.go -------------------------------------------------------------------------------- /pkg/docker/dockertest/ref_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/dockertest/ref_test.go -------------------------------------------------------------------------------- /pkg/docker/dockertest/testdata/alpine.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/dockertest/testdata/alpine.tar -------------------------------------------------------------------------------- /pkg/docker/dockertest/testdata/create-image-fixtures.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/dockertest/testdata/create-image-fixtures.sh -------------------------------------------------------------------------------- /pkg/docker/env.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/env.go -------------------------------------------------------------------------------- /pkg/docker/errors.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/errors.go -------------------------------------------------------------------------------- /pkg/docker/fast_push.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/fast_push.go -------------------------------------------------------------------------------- /pkg/docker/fast_push_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/fast_push_test.go -------------------------------------------------------------------------------- /pkg/docker/host.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/host.go -------------------------------------------------------------------------------- /pkg/docker/host_unix.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/host_unix.go -------------------------------------------------------------------------------- /pkg/docker/host_windows.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/host_windows.go -------------------------------------------------------------------------------- /pkg/docker/login.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/login.go -------------------------------------------------------------------------------- /pkg/docker/monobase.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/monobase.go -------------------------------------------------------------------------------- /pkg/docker/monobase_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/monobase_test.go -------------------------------------------------------------------------------- /pkg/docker/options.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/options.go -------------------------------------------------------------------------------- /pkg/docker/pipeline_push.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/pipeline_push.go -------------------------------------------------------------------------------- /pkg/docker/pipeline_push_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/pipeline_push_test.go -------------------------------------------------------------------------------- /pkg/docker/push.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/push.go -------------------------------------------------------------------------------- /pkg/docker/push_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/push_test.go -------------------------------------------------------------------------------- /pkg/docker/run.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/run.go -------------------------------------------------------------------------------- /pkg/docker/run_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/run_test.go -------------------------------------------------------------------------------- /pkg/docker/standard_push.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/standard_push.go -------------------------------------------------------------------------------- /pkg/docker/standard_push_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/docker/standard_push_test.go -------------------------------------------------------------------------------- /pkg/dockercontext/build_tempdir.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/dockercontext/build_tempdir.go -------------------------------------------------------------------------------- /pkg/dockercontext/build_tempdir_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/dockercontext/build_tempdir_test.go -------------------------------------------------------------------------------- /pkg/dockercontext/directories.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/dockercontext/directories.go -------------------------------------------------------------------------------- /pkg/dockerfile/base.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/dockerfile/base.go -------------------------------------------------------------------------------- /pkg/dockerfile/base_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/dockerfile/base_test.go -------------------------------------------------------------------------------- /pkg/dockerfile/cog_embed.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/dockerfile/cog_embed.go -------------------------------------------------------------------------------- /pkg/dockerfile/cog_embed_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/dockerfile/cog_embed_test.go -------------------------------------------------------------------------------- /pkg/dockerfile/env.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/dockerfile/env.go -------------------------------------------------------------------------------- /pkg/dockerfile/fast_generator.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/dockerfile/fast_generator.go -------------------------------------------------------------------------------- /pkg/dockerfile/fast_generator_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/dockerfile/fast_generator_test.go -------------------------------------------------------------------------------- /pkg/dockerfile/generator.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/dockerfile/generator.go -------------------------------------------------------------------------------- /pkg/dockerfile/generator_factory.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/dockerfile/generator_factory.go -------------------------------------------------------------------------------- /pkg/dockerfile/generator_factory_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/dockerfile/generator_factory_test.go -------------------------------------------------------------------------------- /pkg/dockerfile/monobase_matrix.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/dockerfile/monobase_matrix.go -------------------------------------------------------------------------------- /pkg/dockerfile/monobase_matrix_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/dockerfile/monobase_matrix_test.go -------------------------------------------------------------------------------- /pkg/dockerfile/standard_generator.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/dockerfile/standard_generator.go -------------------------------------------------------------------------------- /pkg/dockerfile/standard_generator_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/dockerfile/standard_generator_test.go -------------------------------------------------------------------------------- /pkg/dockerfile/version_check.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/dockerfile/version_check.go -------------------------------------------------------------------------------- /pkg/dockerignore/dockerignore.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/dockerignore/dockerignore.go -------------------------------------------------------------------------------- /pkg/dockerignore/dockerignore_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/dockerignore/dockerignore_test.go -------------------------------------------------------------------------------- /pkg/env/env.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/env/env.go -------------------------------------------------------------------------------- /pkg/env/env_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/env/env_test.go -------------------------------------------------------------------------------- /pkg/errors/common.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/errors/common.go -------------------------------------------------------------------------------- /pkg/errors/errors.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/errors/errors.go -------------------------------------------------------------------------------- /pkg/global/global.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/global/global.go -------------------------------------------------------------------------------- /pkg/http/client.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/http/client.go -------------------------------------------------------------------------------- /pkg/http/client_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/http/client_test.go -------------------------------------------------------------------------------- /pkg/http/transport.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/http/transport.go -------------------------------------------------------------------------------- /pkg/http/transport_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/http/transport_test.go -------------------------------------------------------------------------------- /pkg/http/user_agent.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/http/user_agent.go -------------------------------------------------------------------------------- /pkg/http/user_agent_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/http/user_agent_test.go -------------------------------------------------------------------------------- /pkg/image/build.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/image/build.go -------------------------------------------------------------------------------- /pkg/image/build_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/image/build_test.go -------------------------------------------------------------------------------- /pkg/image/config.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/image/config.go -------------------------------------------------------------------------------- /pkg/image/model_dependencies.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/image/model_dependencies.go -------------------------------------------------------------------------------- /pkg/image/openapi_schema.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/image/openapi_schema.go -------------------------------------------------------------------------------- /pkg/image/pip_freeze.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/image/pip_freeze.go -------------------------------------------------------------------------------- /pkg/migrate/factory.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/migrate/factory.go -------------------------------------------------------------------------------- /pkg/migrate/factory_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/migrate/factory_test.go -------------------------------------------------------------------------------- /pkg/migrate/migrations.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/migrate/migrations.go -------------------------------------------------------------------------------- /pkg/migrate/migrations_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/migrate/migrations_test.go -------------------------------------------------------------------------------- /pkg/migrate/migrator.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/migrate/migrator.go -------------------------------------------------------------------------------- /pkg/migrate/migrator_v1_v1fast.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/migrate/migrator_v1_v1fast.go -------------------------------------------------------------------------------- /pkg/migrate/migrator_v1_v1fast_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/migrate/migrator_v1_v1fast_test.go -------------------------------------------------------------------------------- /pkg/monobeam/client.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/monobeam/client.go -------------------------------------------------------------------------------- /pkg/monobeam/client_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/monobeam/client_test.go -------------------------------------------------------------------------------- /pkg/path/path.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/path/path.go -------------------------------------------------------------------------------- /pkg/path/path_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/path/path_test.go -------------------------------------------------------------------------------- /pkg/predict/api.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/predict/api.go -------------------------------------------------------------------------------- /pkg/predict/input.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/predict/input.go -------------------------------------------------------------------------------- /pkg/predict/predictor.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/predict/predictor.go -------------------------------------------------------------------------------- /pkg/procedure/validate.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/procedure/validate.go -------------------------------------------------------------------------------- /pkg/procedure/validate_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/procedure/validate_test.go -------------------------------------------------------------------------------- /pkg/registry/client.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/registry/client.go -------------------------------------------------------------------------------- /pkg/registry/client_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/registry/client_test.go -------------------------------------------------------------------------------- /pkg/registry/manifest_result.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/registry/manifest_result.go -------------------------------------------------------------------------------- /pkg/registry/registry_client.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/registry/registry_client.go -------------------------------------------------------------------------------- /pkg/registry/registrytest/mock_client.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/registry/registrytest/mock_client.go -------------------------------------------------------------------------------- /pkg/registry_testhelpers/registry_container.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/registry_testhelpers/registry_container.go -------------------------------------------------------------------------------- /pkg/registry_testhelpers/testdata/docker/registry/v2/blobs/sha256/1c/1c4eef651f65e2f7daee7ee785882ac164b02b78fb74503052a26dc061c90474/data: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/registry_testhelpers/testdata/docker/registry/v2/blobs/sha256/1c/1c4eef651f65e2f7daee7ee785882ac164b02b78fb74503052a26dc061c90474/data -------------------------------------------------------------------------------- /pkg/registry_testhelpers/testdata/docker/registry/v2/blobs/sha256/6e/6e771e15690e2fabf2332d3a3b744495411d6e0b00b2aea64419b58b0066cf81/data: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/registry_testhelpers/testdata/docker/registry/v2/blobs/sha256/6e/6e771e15690e2fabf2332d3a3b744495411d6e0b00b2aea64419b58b0066cf81/data -------------------------------------------------------------------------------- /pkg/registry_testhelpers/testdata/docker/registry/v2/blobs/sha256/75/757d680068d77be46fd1ea20fb21db16f150468c5e7079a08a2e4705aec096ac/data: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/registry_testhelpers/testdata/docker/registry/v2/blobs/sha256/75/757d680068d77be46fd1ea20fb21db16f150468c5e7079a08a2e4705aec096ac/data -------------------------------------------------------------------------------- /pkg/registry_testhelpers/testdata/docker/registry/v2/blobs/sha256/8d/8d591b0b7dea080ea3be9e12ae563eebf9869168ffced1cb25b2470a3d9fe15e/data: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/registry_testhelpers/testdata/docker/registry/v2/blobs/sha256/8d/8d591b0b7dea080ea3be9e12ae563eebf9869168ffced1cb25b2470a3d9fe15e/data -------------------------------------------------------------------------------- /pkg/registry_testhelpers/testdata/docker/registry/v2/blobs/sha256/9a/9a0ff41dccad7a96f324a4655a715c623ed3511c7336361ffa9dadcecbdb99e5/data: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/registry_testhelpers/testdata/docker/registry/v2/blobs/sha256/9a/9a0ff41dccad7a96f324a4655a715c623ed3511c7336361ffa9dadcecbdb99e5/data -------------------------------------------------------------------------------- /pkg/registry_testhelpers/testdata/docker/registry/v2/blobs/sha256/ad/aded1e1a5b3705116fa0a92ba074a5e0b0031647d9c315983ccba2ee5428ec8b/data: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/registry_testhelpers/testdata/docker/registry/v2/blobs/sha256/ad/aded1e1a5b3705116fa0a92ba074a5e0b0031647d9c315983ccba2ee5428ec8b/data -------------------------------------------------------------------------------- /pkg/registry_testhelpers/testdata/docker/registry/v2/blobs/sha256/f1/f18232174bc91741fdf3da96d85011092101a032a93a388b79e99e69c2d5c870/data: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/registry_testhelpers/testdata/docker/registry/v2/blobs/sha256/f1/f18232174bc91741fdf3da96d85011092101a032a93a388b79e99e69c2d5c870/data -------------------------------------------------------------------------------- /pkg/registry_testhelpers/testdata/docker/registry/v2/repositories/alpine/_layers/sha256/6e771e15690e2fabf2332d3a3b744495411d6e0b00b2aea64419b58b0066cf81/link: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/registry_testhelpers/testdata/docker/registry/v2/repositories/alpine/_layers/sha256/6e771e15690e2fabf2332d3a3b744495411d6e0b00b2aea64419b58b0066cf81/link -------------------------------------------------------------------------------- /pkg/registry_testhelpers/testdata/docker/registry/v2/repositories/alpine/_layers/sha256/8d591b0b7dea080ea3be9e12ae563eebf9869168ffced1cb25b2470a3d9fe15e/link: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/registry_testhelpers/testdata/docker/registry/v2/repositories/alpine/_layers/sha256/8d591b0b7dea080ea3be9e12ae563eebf9869168ffced1cb25b2470a3d9fe15e/link -------------------------------------------------------------------------------- /pkg/registry_testhelpers/testdata/docker/registry/v2/repositories/alpine/_layers/sha256/aded1e1a5b3705116fa0a92ba074a5e0b0031647d9c315983ccba2ee5428ec8b/link: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/registry_testhelpers/testdata/docker/registry/v2/repositories/alpine/_layers/sha256/aded1e1a5b3705116fa0a92ba074a5e0b0031647d9c315983ccba2ee5428ec8b/link -------------------------------------------------------------------------------- /pkg/registry_testhelpers/testdata/docker/registry/v2/repositories/alpine/_layers/sha256/f18232174bc91741fdf3da96d85011092101a032a93a388b79e99e69c2d5c870/link: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/registry_testhelpers/testdata/docker/registry/v2/repositories/alpine/_layers/sha256/f18232174bc91741fdf3da96d85011092101a032a93a388b79e99e69c2d5c870/link -------------------------------------------------------------------------------- /pkg/registry_testhelpers/testdata/docker/registry/v2/repositories/alpine/_manifests/revisions/sha256/1c4eef651f65e2f7daee7ee785882ac164b02b78fb74503052a26dc061c90474/link: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/registry_testhelpers/testdata/docker/registry/v2/repositories/alpine/_manifests/revisions/sha256/1c4eef651f65e2f7daee7ee785882ac164b02b78fb74503052a26dc061c90474/link -------------------------------------------------------------------------------- /pkg/registry_testhelpers/testdata/docker/registry/v2/repositories/alpine/_manifests/revisions/sha256/757d680068d77be46fd1ea20fb21db16f150468c5e7079a08a2e4705aec096ac/link: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/registry_testhelpers/testdata/docker/registry/v2/repositories/alpine/_manifests/revisions/sha256/757d680068d77be46fd1ea20fb21db16f150468c5e7079a08a2e4705aec096ac/link -------------------------------------------------------------------------------- /pkg/registry_testhelpers/testdata/docker/registry/v2/repositories/alpine/_manifests/revisions/sha256/9a0ff41dccad7a96f324a4655a715c623ed3511c7336361ffa9dadcecbdb99e5/link: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/registry_testhelpers/testdata/docker/registry/v2/repositories/alpine/_manifests/revisions/sha256/9a0ff41dccad7a96f324a4655a715c623ed3511c7336361ffa9dadcecbdb99e5/link -------------------------------------------------------------------------------- /pkg/registry_testhelpers/testdata/docker/registry/v2/repositories/alpine/_manifests/tags/latest/current/link: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/registry_testhelpers/testdata/docker/registry/v2/repositories/alpine/_manifests/tags/latest/current/link -------------------------------------------------------------------------------- /pkg/registry_testhelpers/testdata/docker/registry/v2/repositories/alpine/_manifests/tags/latest/index/sha256/9a0ff41dccad7a96f324a4655a715c623ed3511c7336361ffa9dadcecbdb99e5/link: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/registry_testhelpers/testdata/docker/registry/v2/repositories/alpine/_manifests/tags/latest/index/sha256/9a0ff41dccad7a96f324a4655a715c623ed3511c7336361ffa9dadcecbdb99e5/link -------------------------------------------------------------------------------- /pkg/requirements/requirements.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/requirements/requirements.go -------------------------------------------------------------------------------- /pkg/requirements/requirements_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/requirements/requirements_test.go -------------------------------------------------------------------------------- /pkg/update/state.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/update/state.go -------------------------------------------------------------------------------- /pkg/update/update.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/update/update.go -------------------------------------------------------------------------------- /pkg/util/console/console.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/util/console/console.go -------------------------------------------------------------------------------- /pkg/util/console/formatting.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/util/console/formatting.go -------------------------------------------------------------------------------- /pkg/util/console/global.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/util/console/global.go -------------------------------------------------------------------------------- /pkg/util/console/interactive.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/util/console/interactive.go -------------------------------------------------------------------------------- /pkg/util/console/levels.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/util/console/levels.go -------------------------------------------------------------------------------- /pkg/util/console/term.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/util/console/term.go -------------------------------------------------------------------------------- /pkg/util/env.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/util/env.go -------------------------------------------------------------------------------- /pkg/util/errors.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/util/errors.go -------------------------------------------------------------------------------- /pkg/util/files/files.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/util/files/files.go -------------------------------------------------------------------------------- /pkg/util/files/files_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/util/files/files_test.go -------------------------------------------------------------------------------- /pkg/util/hash.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/util/hash.go -------------------------------------------------------------------------------- /pkg/util/hash_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/util/hash_test.go -------------------------------------------------------------------------------- /pkg/util/mime/mime.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/util/mime/mime.go -------------------------------------------------------------------------------- /pkg/util/mime/mime_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/util/mime/mime_test.go -------------------------------------------------------------------------------- /pkg/util/net.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/util/net.go -------------------------------------------------------------------------------- /pkg/util/overwrite_yaml.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/util/overwrite_yaml.go -------------------------------------------------------------------------------- /pkg/util/overwrite_yaml_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/util/overwrite_yaml_test.go -------------------------------------------------------------------------------- /pkg/util/platform.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/util/platform.go -------------------------------------------------------------------------------- /pkg/util/ringbuffer.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/util/ringbuffer.go -------------------------------------------------------------------------------- /pkg/util/shell/net.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/util/shell/net.go -------------------------------------------------------------------------------- /pkg/util/shell/pipes.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/util/shell/pipes.go -------------------------------------------------------------------------------- /pkg/util/slices/slices.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/util/slices/slices.go -------------------------------------------------------------------------------- /pkg/util/version/version.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/util/version/version.go -------------------------------------------------------------------------------- /pkg/util/version/version_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/util/version/version_test.go -------------------------------------------------------------------------------- /pkg/web/client.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/web/client.go -------------------------------------------------------------------------------- /pkg/web/client_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/web/client_test.go -------------------------------------------------------------------------------- /pkg/weights/fast_weights.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/weights/fast_weights.go -------------------------------------------------------------------------------- /pkg/weights/fast_weights_manifest.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/weights/fast_weights_manifest.go -------------------------------------------------------------------------------- /pkg/weights/fast_weights_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/weights/fast_weights_test.go -------------------------------------------------------------------------------- /pkg/weights/manifest.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/weights/manifest.go -------------------------------------------------------------------------------- /pkg/weights/weights.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/weights/weights.go -------------------------------------------------------------------------------- /pkg/weights/weights_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pkg/weights/weights_test.go -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/pyproject.toml -------------------------------------------------------------------------------- /python/cog/.gitignore: -------------------------------------------------------------------------------- 1 | /_version.py 2 | -------------------------------------------------------------------------------- /python/cog/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/__init__.py -------------------------------------------------------------------------------- /python/cog/base_input.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/base_input.py -------------------------------------------------------------------------------- /python/cog/base_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/base_predictor.py -------------------------------------------------------------------------------- /python/cog/code_xforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/code_xforms.py -------------------------------------------------------------------------------- /python/cog/command/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /python/cog/command/ast_openapi_schema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/command/ast_openapi_schema.py -------------------------------------------------------------------------------- /python/cog/command/call_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/command/call_graph.py -------------------------------------------------------------------------------- /python/cog/command/migrate_v1_v1fast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/command/migrate_v1_v1fast.py -------------------------------------------------------------------------------- /python/cog/command/openapi_schema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/command/openapi_schema.py -------------------------------------------------------------------------------- /python/cog/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/config.py -------------------------------------------------------------------------------- /python/cog/env_property.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/env_property.py -------------------------------------------------------------------------------- /python/cog/errors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/errors.py -------------------------------------------------------------------------------- /python/cog/ext/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/ext/__init__.py -------------------------------------------------------------------------------- /python/cog/files.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/files.py -------------------------------------------------------------------------------- /python/cog/json.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/json.py -------------------------------------------------------------------------------- /python/cog/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/logging.py -------------------------------------------------------------------------------- /python/cog/mimetypes_ext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/mimetypes_ext.py -------------------------------------------------------------------------------- /python/cog/mode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/mode.py -------------------------------------------------------------------------------- /python/cog/predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/predictor.py -------------------------------------------------------------------------------- /python/cog/schema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/schema.py -------------------------------------------------------------------------------- /python/cog/server/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /python/cog/server/connection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/server/connection.py -------------------------------------------------------------------------------- /python/cog/server/errors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/server/errors.py -------------------------------------------------------------------------------- /python/cog/server/eventtypes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/server/eventtypes.py -------------------------------------------------------------------------------- /python/cog/server/exceptions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/server/exceptions.py -------------------------------------------------------------------------------- /python/cog/server/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/server/helpers.py -------------------------------------------------------------------------------- /python/cog/server/http.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/server/http.py -------------------------------------------------------------------------------- /python/cog/server/probes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/server/probes.py -------------------------------------------------------------------------------- /python/cog/server/response_throttler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/server/response_throttler.py -------------------------------------------------------------------------------- /python/cog/server/runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/server/runner.py -------------------------------------------------------------------------------- /python/cog/server/scope.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/server/scope.py -------------------------------------------------------------------------------- /python/cog/server/telemetry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/server/telemetry.py -------------------------------------------------------------------------------- /python/cog/server/useragent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/server/useragent.py -------------------------------------------------------------------------------- /python/cog/server/webhook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/server/webhook.py -------------------------------------------------------------------------------- /python/cog/server/worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/server/worker.py -------------------------------------------------------------------------------- /python/cog/suppress_output.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/suppress_output.py -------------------------------------------------------------------------------- /python/cog/types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/types.py -------------------------------------------------------------------------------- /python/cog/wait.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/cog/wait.py -------------------------------------------------------------------------------- /python/tests/cog/test_files.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/cog/test_files.py -------------------------------------------------------------------------------- /python/tests/command/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /python/tests/command/call_graph_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/command/call_graph_test.py -------------------------------------------------------------------------------- /python/tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/conftest.py -------------------------------------------------------------------------------- /python/tests/fixtures/argv_override.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/fixtures/argv_override.py -------------------------------------------------------------------------------- /python/tests/server/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /python/tests/server/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/conftest.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/async_setup_uses_same_loop_as_predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/async_setup_uses_same_loop_as_predict.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/catch_in_predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/catch_in_predict.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/complex_output.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/complex_output.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/count_up.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/count_up.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/exc_in_predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/exc_in_predict.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/exc_in_setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/exc_in_setup.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/exc_in_setup_and_predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/exc_in_setup_and_predict.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/exc_on_import.py: -------------------------------------------------------------------------------- 1 | raise RuntimeException("this should not be importable") 2 | -------------------------------------------------------------------------------- /python/tests/server/fixtures/exit_in_predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/exit_in_predict.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/exit_in_setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/exit_in_setup.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/exit_on_import.py: -------------------------------------------------------------------------------- 1 | import sys 2 | 3 | sys.exit() 4 | -------------------------------------------------------------------------------- /python/tests/server/fixtures/function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/function.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/hello_world.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/hello_world.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/hello_world_async.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/hello_world_async.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/import_err.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/import_err.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/input_choices.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/input_choices.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/input_choices_integer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/input_choices_integer.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/input_choices_iterable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/input_choices_iterable.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/input_deprecated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/input_deprecated.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/input_file.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/input_file.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/input_ge_le.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/input_ge_le.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/input_integer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/input_integer.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/input_integer_default.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/input_integer_default.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/input_kwargs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/input_kwargs.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/input_literal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/input_literal.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/input_literal_integer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/input_literal_integer.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/input_multiple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/input_multiple.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/input_none.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/input_none.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/input_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/input_path.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/input_path_2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/input_path_2.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/input_path_or_none.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/input_path_or_none.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/input_secret.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/input_secret.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/input_string.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/input_string.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/input_union_integer_or_list_of_integers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/input_union_integer_or_list_of_integers.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/input_union_string_or_list_of_strings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/input_union_string_or_list_of_strings.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/input_unsupported_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/input_unsupported_type.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/input_untyped.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/input_untyped.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/killed_in_predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/killed_in_predict.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/logging.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/logging_async.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/logging_async.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/missing_predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/missing_predict.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/missing_predictor.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /python/tests/server/fixtures/openapi_complex_input.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/openapi_complex_input.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/openapi_custom_output_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/openapi_custom_output_type.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/openapi_input_int_choices.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/openapi_input_int_choices.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/openapi_optional_output_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/openapi_optional_output_type.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/openapi_output_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/openapi_output_list.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/openapi_output_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/openapi_output_type.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/openapi_output_yield.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/openapi_output_yield.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/output_complex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/output_complex.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/output_file.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/output_file.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/output_file_named.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/output_file_named.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/output_iterator_complex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/output_iterator_complex.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/output_numpy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/output_numpy.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/output_path_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/output_path_image.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/output_path_text.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/output_path_text.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/output_wrong_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/output_wrong_type.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/record_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/record_metric.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/record_metric_async.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/record_metric_async.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/setup.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/setup_async.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/setup_async.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/setup_async_with_sync_predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/setup_async_with_sync_predict.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/setup_uses_async.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/setup_uses_async.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/setup_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/setup_weights.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/simple.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/simple_async.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/simple_async.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/sleep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/sleep.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/sleep_async.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/sleep_async.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/slow_predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/slow_predict.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/slow_setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/slow_setup.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/steps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/steps.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/stream_redirector_race_condition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/stream_redirector_race_condition.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/train.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/with_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/with_context.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/with_context_async.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/with_context_async.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/yield_concatenate_iterator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/yield_concatenate_iterator.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/yield_files.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/yield_files.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/yield_strings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/yield_strings.py -------------------------------------------------------------------------------- /python/tests/server/fixtures/yield_strings_file_input.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/fixtures/yield_strings_file_input.py -------------------------------------------------------------------------------- /python/tests/server/test_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/test_helpers.py -------------------------------------------------------------------------------- /python/tests/server/test_http.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/test_http.py -------------------------------------------------------------------------------- /python/tests/server/test_http_input.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/test_http_input.py -------------------------------------------------------------------------------- /python/tests/server/test_http_output.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/test_http_output.py -------------------------------------------------------------------------------- /python/tests/server/test_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/test_predictor.py -------------------------------------------------------------------------------- /python/tests/server/test_probes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/test_probes.py -------------------------------------------------------------------------------- /python/tests/server/test_response_throttler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/test_response_throttler.py -------------------------------------------------------------------------------- /python/tests/server/test_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/test_runner.py -------------------------------------------------------------------------------- /python/tests/server/test_webhook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/test_webhook.py -------------------------------------------------------------------------------- /python/tests/server/test_worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/server/test_worker.py -------------------------------------------------------------------------------- /python/tests/test_code_xforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/test_code_xforms.py -------------------------------------------------------------------------------- /python/tests/test_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/test_config.py -------------------------------------------------------------------------------- /python/tests/test_json.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/test_json.py -------------------------------------------------------------------------------- /python/tests/test_mimetypes_ext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/test_mimetypes_ext.py -------------------------------------------------------------------------------- /python/tests/test_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/test_predictor.py -------------------------------------------------------------------------------- /python/tests/test_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/test_types.py -------------------------------------------------------------------------------- /python/tests/test_wait.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/python/tests/test_wait.py -------------------------------------------------------------------------------- /script/format: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/script/format -------------------------------------------------------------------------------- /script/lint: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/script/lint -------------------------------------------------------------------------------- /script/setup: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/script/setup -------------------------------------------------------------------------------- /script/test: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/script/test -------------------------------------------------------------------------------- /script/test-all: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/script/test-all -------------------------------------------------------------------------------- /script/test-go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/script/test-go -------------------------------------------------------------------------------- /script/test-python: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/script/test-python -------------------------------------------------------------------------------- /test-integration/test_integration/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test-integration/test_integration/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/conftest.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/apt-packages/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/apt-packages/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/apt-packages/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/apt-packages/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/async-sleep-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/async-sleep-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/async-sleep-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/async-sleep-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/async-string-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/async-string-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/async-string-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/async-string-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/bad-dockerignore/.dockerignore: -------------------------------------------------------------------------------- 1 | .cog 2 | -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/bad-dockerignore/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/bad-dockerignore/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/bad-dockerignore/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/bad-dockerignore/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/cog-runtime-float/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/cog-runtime-float/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/cog-runtime-float/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/cog-runtime-float/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/cog-runtime-int/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/cog-runtime-int/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/cog-runtime-int/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/cog-runtime-int/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/complex-types-list-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/complex-types-list-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/complex-types-list-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/complex-types-list-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/complex-types-list-project/requirements.txt: -------------------------------------------------------------------------------- 1 | anthropic[vertex]==0.45.2 2 | -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/complex-types/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/complex-types/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/complex-types/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/complex-types/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/complex-types/requirements.txt: -------------------------------------------------------------------------------- 1 | anthropic[vertex]==0.45.2 2 | -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/complex_output_project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/complex_output_project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/complex_output_project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/complex_output_project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/env-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/env-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/env-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/env-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/fast-build/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/fast-build/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/fast-build/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/fast-build/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/fast-build/requirements.txt: -------------------------------------------------------------------------------- 1 | torch==2.5.0 2 | beautifulsoup4==4.12.3 3 | -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/ffmpeg-package/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/ffmpeg-package/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/ffmpeg-package/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/ffmpeg-package/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/file-input-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/file-input-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/file-input-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/file-input-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/file-input-project/test.txt: -------------------------------------------------------------------------------- 1 | text 2 | -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/file-list-input-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/file-list-input-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/file-list-input-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/file-list-input-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/future-annotations-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/future-annotations-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/future-annotations-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/future-annotations-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/future-annotations-project/some_image.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/future-annotations-project/some_image.jpg -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/glb-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/glb-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/glb-project/mesh.glb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/glb-project/mesh.glb -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/glb-project/output.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/glb-project/output.txt -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/glb-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/glb-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/granite-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/granite-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/granite-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/granite-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/granite-project/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/granite-project/requirements.txt -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/install-requires-packaging/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/install-requires-packaging/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/install-requires-packaging/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/install-requires-packaging/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/install-requires-packaging/requirements.txt: -------------------------------------------------------------------------------- 1 | causal-conv1d==1.5.0.post8 2 | torch==2.5.0 3 | -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/int-none-output-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/int-none-output-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/int-none-output-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/int-none-output-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/int-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/int-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/int-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/int-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/invalid-int-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/invalid-int-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/invalid-int-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/invalid-int-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/local-whl-install/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/local-whl-install/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/local-whl-install/my_package-0.1-py2.py3-none-any.whl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/local-whl-install/my_package-0.1-py2.py3-none-any.whl -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/local-whl-install/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/local-whl-install/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/local-whl-install/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/local-whl-install/requirements.txt -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/many-inputs-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/many-inputs-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/many-inputs-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/many-inputs-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/migration-gpu-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/migration-gpu-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/migration-gpu-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/migration-gpu-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/migration-no-python-changes-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/migration-no-python-changes-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/migration-no-python-changes-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/migration-no-python-changes-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/migration-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/migration-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/migration-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/migration-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/new-union-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/new-union-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/new-union-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/new-union-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/no-predictor-project/cog.yaml: -------------------------------------------------------------------------------- 1 | build: 2 | python_version: 3.9 3 | -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/optional-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/optional-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/optional-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/optional-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/overrides-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/overrides-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/overrides-project/overrides.txt: -------------------------------------------------------------------------------- 1 | numpy==1.26.4 2 | -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/overrides-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/overrides-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/overrides-project/requirements.txt: -------------------------------------------------------------------------------- 1 | numpy==1.26.3 2 | -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/path-input-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/path-input-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/path-input-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/path-input-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/path-list-input-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/path-list-input-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/path-list-input-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/path-list-input-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/path-list-output-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/path-list-output-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/path-list-output-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/path-list-output-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/path-output-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/path-output-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/path-output-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/path-output-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/path-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/path-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/path-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/path-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/pipeline-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/pipeline-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/pipeline-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/pipeline-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/pipeline-requirements-project/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/pipeline-requirements-project/.gitignore -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/pipeline-requirements-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/pipeline-requirements-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/pipeline-requirements-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/pipeline-requirements-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/procedure-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/procedure-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/procedure-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/procedure-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/pydantic1-none/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/pydantic1-none/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/pydantic1-none/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/pydantic1-none/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/pydantic2-output/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/pydantic2-output/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/pydantic2-output/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/pydantic2-output/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/pydantic2-output/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/pydantic2-output/train.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/pydantic2/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/pydantic2/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/pydantic2/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/pydantic2/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/python-313/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/python-313/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/python-313/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/python-313/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/python_37/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/python_37/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/python_37/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/python_37/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/secrets-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/secrets-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/secrets-project/file-secret.txt: -------------------------------------------------------------------------------- 1 | file_secret_value 2 | -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/secrets-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/secrets-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/setup-subprocess-double-fork-http-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/setup-subprocess-double-fork-http-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/setup-subprocess-double-fork-http-project/pong.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/setup-subprocess-double-fork-http-project/pong.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/setup-subprocess-double-fork-http-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/setup-subprocess-double-fork-http-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/setup-subprocess-double-fork-http-project/run-pong.sh: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env bash 2 | python ./pong.py & 3 | wait 4 | -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/setup-subprocess-double-fork-project/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/setup-subprocess-double-fork-project/.gitignore -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/setup-subprocess-double-fork-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/setup-subprocess-double-fork-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/setup-subprocess-double-fork-project/forker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/setup-subprocess-double-fork-project/forker.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/setup-subprocess-double-fork-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/setup-subprocess-double-fork-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/setup-subprocess-double-fork-project/run-forker.sh: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env bash 2 | python ./forker.py & 3 | wait 4 | -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/setup-subprocess-multiprocessing-project/.gitignore: -------------------------------------------------------------------------------- 1 | *.tmp 2 | -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/setup-subprocess-multiprocessing-project/bg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/setup-subprocess-multiprocessing-project/bg.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/setup-subprocess-multiprocessing-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/setup-subprocess-multiprocessing-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/setup-subprocess-multiprocessing-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/setup-subprocess-multiprocessing-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/setup-subprocess-simple-project/child.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/setup-subprocess-simple-project/child.sh -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/setup-subprocess-simple-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/setup-subprocess-simple-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/setup-subprocess-simple-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/setup-subprocess-simple-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/string-list-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/string-list-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/string-list-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/string-list-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/string-none-output-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/string-none-output-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/string-none-output-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/string-none-output-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/string-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/string-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/string-project/input.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/string-project/input.json -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/string-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/string-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/subdirectory-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/subdirectory-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/subdirectory-project/my-subdir/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/subdirectory-project/my-subdir/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/subdirectory-project/mylib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/subdirectory-project/mylib.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/tensorflow-project/.requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/tensorflow-project/.requirements.txt -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/tensorflow-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/tensorflow-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/tensorflow-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/tensorflow-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/torch-270-cuda-126/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/torch-270-cuda-126/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/torch-270-cuda-126/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/torch-270-cuda-126/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/torch-271-cuda-128/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/torch-271-cuda-128/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/torch-271-cuda-128/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/torch-271-cuda-128/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/torch-baseimage-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/torch-baseimage-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/torch-baseimage-project/openapi.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/torch-baseimage-project/openapi.json -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/torch-baseimage-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/torch-baseimage-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/torch-cuda-baseimage-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/torch-cuda-baseimage-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/torch-cuda-baseimage-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/torch-cuda-baseimage-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/train-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/train-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/train-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/train-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/train-project/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/train-project/train.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/training-setup-project/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/training-setup-project/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/training-setup-project/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/training-setup-project/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/training-setup-project/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/training-setup-project/train.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/zsh-package/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/zsh-package/cog.yaml -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/zsh-package/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/fixtures/zsh-package/predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/fixtures/zsh-package/requirements.txt: -------------------------------------------------------------------------------- 1 | torch==2.5.0 2 | -------------------------------------------------------------------------------- /test-integration/test_integration/test_build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/test_build.py -------------------------------------------------------------------------------- /test-integration/test_integration/test_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/test_config.py -------------------------------------------------------------------------------- /test-integration/test_integration/test_migrate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/test_migrate.py -------------------------------------------------------------------------------- /test-integration/test_integration/test_predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/test_predict.py -------------------------------------------------------------------------------- /test-integration/test_integration/test_run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/test_run.py -------------------------------------------------------------------------------- /test-integration/test_integration/test_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/test_train.py -------------------------------------------------------------------------------- /test-integration/test_integration/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/test-integration/test_integration/util.py -------------------------------------------------------------------------------- /tools/compatgen/internal/cuda.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/tools/compatgen/internal/cuda.go -------------------------------------------------------------------------------- /tools/compatgen/internal/tensorflow.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/tools/compatgen/internal/tensorflow.go -------------------------------------------------------------------------------- /tools/compatgen/internal/torch.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/tools/compatgen/internal/torch.go -------------------------------------------------------------------------------- /tools/compatgen/internal/torch_package.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/tools/compatgen/internal/torch_package.go -------------------------------------------------------------------------------- /tools/compatgen/internal/torch_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/tools/compatgen/internal/torch_test.go -------------------------------------------------------------------------------- /tools/compatgen/internal/torch_test.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/tools/compatgen/internal/torch_test.html -------------------------------------------------------------------------------- /tools/compatgen/internal/util.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/tools/compatgen/internal/util.go -------------------------------------------------------------------------------- /tools/compatgen/main.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/tools/compatgen/main.go -------------------------------------------------------------------------------- /tools/install.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/tools/install.sh -------------------------------------------------------------------------------- /tools/test-registry-util/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/tools/test-registry-util/README.md -------------------------------------------------------------------------------- /tools/test-registry-util/main.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/tools/test-registry-util/main.go -------------------------------------------------------------------------------- /tools/uploader/iface.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/tools/uploader/iface.go -------------------------------------------------------------------------------- /tools/uploader/s3.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/tools/uploader/s3.go -------------------------------------------------------------------------------- /tox.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/tox.ini -------------------------------------------------------------------------------- /uv.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/cog/HEAD/uv.lock --------------------------------------------------------------------------------