/model-hosting-container-standards

Primary LanguagePythonApache License 2.0Apache-2.0

Model Hosting Container Standards

Python toolkit for standardized model hosting container implementations with Amazon SageMaker integration.

Overview

This repository provides a Python toolkit that enables TensorRT-LLM and vLLM integration with Amazon SageMaker hosting platform for efficient model deployment and inference.

Repository Structure

ModelHostingContainerStandards/
├── python/                    # Python implementation
│   ├── model_hosting_container_standards/  # Main Python package
│   │   ├── __init__.py
│   │   ├── config.py
│   │   ├── logging_config.py
│   │   ├── utils.py
│   │   ├── common/            # Common utilities
│   │   │   ├── fastapi/       # FastAPI integration
│   │   │   ├── custom_code_ref_resolver/  # Dynamic code loading
│   │   │   └── handler/       # Handler specifications
│   │   └── sagemaker/         # SageMaker integration
│   │       └── lora/          # LoRA adapter support
│   ├── tests/                 # Package tests
│   ├── examples/              # Python examples and demos
│   ├── pyproject.toml         # Python project configuration
│   ├── Makefile               # Build automation
│   └── README.md              # Python-specific documentation
├── docs/                      # Documentation
├── examples/                  # Top-level examples
├── .github/                   # GitHub templates and workflows
├── Config                     # Shared configuration files
└── README.md                  # This file

Quick Start

cd python
poetry install
poetry shell

See the Python README for detailed usage instructions, examples, and development workflow.

Contributing

When contributing to this repository:

  1. Place Python-specific code in the python/ directory
  2. Follow the established patterns for project structure
  3. Include tests for new functionality
  4. Update documentation as needed
  5. Run pre-commit hooks to ensure code quality

Security

See CONTRIBUTING for more information.

License

This project is licensed under the Apache-2.0 License.