diff --git a/CHANGELOG.md b/CHANGELOG.md index c51a2254..99139b72 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -1,3 +1,10 @@ +## [[11.0.2]](https://github.com/Clarifai/clarifai-python/releases/tag/11.0.2) - [PyPI](https://pypi.org/project/clarifai/11.0.2/) - 2025-01-14 + +### Changed + + - Update base images [(#481)] (https://github.com/Clarifai/clarifai-python/pull/481) + - Optimize downloads from HF [(#480)] (https://github.com/Clarifai/clarifai-python/pull/480) + ## [[11.0.1]](https://github.com/Clarifai/clarifai-python/releases/tag/11.0.1) - [PyPI](https://pypi.org/project/clarifai/11.0.1/) - 2025-01-14 ### Changed diff --git a/clarifai/__init__.py b/clarifai/__init__.py index b5d09801..41238981 100644 --- a/clarifai/__init__.py +++ b/clarifai/__init__.py @@ -1 +1 @@ -__version__ = "11.0.1" +__version__ = "11.0.2" diff --git a/clarifai/runners/utils/const.py b/clarifai/runners/utils/const.py index 04e04dc5..6b3816e7 100644 --- a/clarifai/runners/utils/const.py +++ b/clarifai/runners/utils/const.py @@ -1,39 +1,41 @@ -PYTHON_BASE_IMAGE = 'public.ecr.aws/clarifai-models/python-base:{python_version}' -TORCH_BASE_IMAGE = 'public.ecr.aws/clarifai-models/torch:{torch_version}-py{python_version}-cuda{cuda_version}' +import os + +registry = os.environ.get('CLARIFAI_BASE_IMAGE_REGISTRY', 'public.ecr.aws/clarifai-models') + +PYTHON_BASE_IMAGE = registry + '/python-base:{python_version}' +TORCH_BASE_IMAGE = registry + '/torch:{torch_version}-py{python_version}-cuda{cuda_version}' # List of available python base images -AVAILABLE_PYTHON_IMAGES = ['3.8', '3.9', '3.10', '3.11', '3.12', '3.13'] +AVAILABLE_PYTHON_IMAGES = ['3.11', '3.12', '3.13'] -DEFAULT_PYTHON_VERSION = 3.11 +DEFAULT_PYTHON_VERSION = 3.12 # List of available torch images AVAILABLE_TORCH_IMAGES = [ - '1.13.1-py3.8-cuda117', - '1.13.1-py3.9-cuda117', - '1.13.1-py3.10-cuda117', - '2.1.2-py3.8-cuda121', - '2.1.2-py3.9-cuda121', - '2.1.2-py3.10-cuda121', - '2.1.2-py3.11-cuda121', - '2.2.2-py3.8-cuda121', - '2.2.2-py3.9-cuda121', - '2.2.2-py3.10-cuda121', '2.2.2-py3.11-cuda121', - '2.2.2-py3.12-cuda121', - '2.3.1-py3.8-cuda121', - '2.3.1-py3.9-cuda121', - '2.3.1-py3.10-cuda121', '2.3.1-py3.11-cuda121', - '2.3.1-py3.12-cuda121', - '2.4.1-py3.8-cuda124', - '2.4.1-py3.9-cuda124', - '2.4.1-py3.10-cuda124', + '2.4.0-py3.11-cuda121', + '2.4.0-py3.11-cuda124', + '2.4.1-py3.11-cuda121', '2.4.1-py3.11-cuda124', - '2.4.1-py3.12-cuda124', - '2.5.1-py3.9-cuda124', - '2.5.1-py3.10-cuda124', + '2.5.1-py3.11-cuda121', '2.5.1-py3.11-cuda124', + '2.2.2-py3.12-cuda121', + '2.3.1-py3.12-cuda121', + '2.4.0-py3.12-cuda121', + '2.4.0-py3.12-cuda124', + '2.4.1-py3.12-cuda121', + '2.4.1-py3.12-cuda124', + '2.5.1-py3.12-cuda121', '2.5.1-py3.12-cuda124', + # '2.2.2-py3.13-cuda121', + # '2.3.1-py3.13-cuda121', + # '2.4.0-py3.13-cuda121', + # '2.4.0-py3.13-cuda124', + # '2.4.1-py3.13-cuda121', + # '2.4.1-py3.13-cuda124', + # '2.5.1-py3.13-cuda121', + # '2.5.1-py3.13-cuda124', ] CONCEPTS_REQUIRED_MODEL_TYPE = [ 'visual-classifier', 'visual-detector', 'visual-segmenter', 'text-classifier'