gpustack-runner 0.1.23.post5__tar.gz → 0.1.24.post1__tar.gz

{gpustack_runner-0.1.23.post5 → gpustack_runner-0.1.24.post1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gpustack-runner
-Version: 0.1.23.post5
+Version: 0.1.24.post1
 Summary: GPUStack Runner is library for registering runnable accelerated backends and services in GPUStack.
 Project-URL: Homepage, https://github.com/gpustack/runner
 Project-URL: Bug Tracker, https://github.com/gpustack/gpustack/issues
@@ -46,24 +46,20 @@ The following table lists the supported accelerated backends and their correspon
 
 ### Ascend CANN
 
-> [!CAUTION]
-> Since v0.1.23:
-> - Deprecated MindIE `2.1.rc1`.
-
 > [!WARNING]
 > - The Atlas 300I series is currently experimental in vLLM, only supporting eager mode and float16 data type. And there
     are some known issues for running vLLM, you can refer to
     vllm-ascend [#3316](https://github.com/vllm-project/vllm-ascend/issues/3316)
     and [#2795](https://github.com/vllm-project/vllm-ascend/issues/2795).
 
-| CANN Version <br/> (Variant) | MindIE                   | vLLM                                                       | SGLang                 |
-|------------------------------|--------------------------|------------------------------------------------------------|------------------------|
-| 8.3 (A3/910C)                | `2.2.rc1`                | `0.13.0`, `0.12.0`, `0.11.0`                               | `0.5.7`, `0.5.6.post2` |
-| 8.3 (910B)                   | `2.2.rc1`                | `0.13.0`, `0.12.0`, `0.11.0`                               | `0.5.7`, `0.5.6.post2` |
-| 8.3 (310P)                   | `2.2.rc1`                |                                                            |                        |
-| 8.2 (A3/910C)                | `2.1.rc2`                | `0.10.2`, `0.10.1.1`                                       | `0.5.2`, `0.5.1.post3` |
-| 8.2 (910B)                   | `2.1.rc2`, ~~`2.1.rc1`~~ | `0.10.2`, `0.10.1.1`, <br/>`0.10.0`, `0.9.2`, <br/>`0.9.1` | `0.5.2`, `0.5.1.post3` |
-| 8.2 (310P)                   | `2.1.rc2`, ~~`2.1.rc1`~~ | `0.10.0`, `0.9.2`                                          |                        |
+| CANN Version <br/> (Variant) | MindIE    | vLLM                                                       | SGLang                 |
+|------------------------------|-----------|------------------------------------------------------------|------------------------|
+| 8.3 (A3/910C)                | `2.2.rc1` | `0.13.0`, `0.12.0`, `0.11.0`                               | `0.5.7`, `0.5.6.post2` |
+| 8.3 (910B)                   | `2.2.rc1` | `0.13.0`, `0.12.0`, `0.11.0`                               | `0.5.7`, `0.5.6.post2` |
+| 8.3 (310P)                   | `2.2.rc1` |                                                            |                        |
+| 8.2 (A3/910C)                | `2.1.rc2` | `0.10.2`, `0.10.1.1`                                       | `0.5.2`, `0.5.1.post3` |
+| 8.2 (910B)                   | `2.1.rc2` | `0.10.2`, `0.10.1.1`, <br/>`0.10.0`, `0.9.2`, <br/>`0.9.1` | `0.5.2`, `0.5.1.post3` |
+| 8.2 (310P)                   | `2.1.rc2` | `0.10.0`, `0.9.2`                                          |                        |
 
 ### Iluvatar CoreX
 
@@ -73,13 +69,6 @@ The following table lists the supported accelerated backends and their correspon
 
 ### NVIDIA CUDA
 
-> [!CAUTION]
-> Since v0.1.23:
-> - Deprecated all services for CUDA 12.4.
-> - Deprecated vLLM `0.11.0`, `0.10.1.1`, `0.10.0`.
-> - Deprecated SGLang `0.5.5`.
-> - Deprecated VoxBox `0.0.20`.
-
 > [!NOTE]
 > - CUDA 12.9 supports Compute Capabilities:
     `7.5 8.0+PTX 8.9 9.0 10.0 10.3 12.0 12.1+PTX`.
@@ -88,12 +77,11 @@ The following table lists the supported accelerated backends and their correspon
 > - CUDA 12.6/12.4 supports Compute Capabilities:
     `7.5 8.0+PTX 8.9 9.0+PTX`.
 
-| CUDA Version <br/> (Variant) | vLLM                                                                                              | SGLang                                                                 | VoxBox                 |
-|------------------------------|---------------------------------------------------------------------------------------------------|------------------------------------------------------------------------|------------------------|
-| 12.9                         | `0.13.0`, `0.12.0`, <br/>`0.11.2`                                                                 | `0.5.7`, `0.5.6.post2`                                                 |                        |
-| 12.8                         | `0.13.0`, `0.12.0`, <br/>`0.11.2`, ~~`0.11.0`~~, <br/>`0.10.2`, ~~`0.10.1.1`~~, <br/>~~`0.10.0`~~ | `0.5.7`, `0.5.6.post2`, `0.5.5.post3`, <br/>~~`0.5.5`~~, `0.5.4.post3` | `0.0.21`, ~~`0.0.20`~~ |
-| 12.6                         | `0.13.0`, `0.12.0`, <br/>`0.11.2`, ~~`0.11.0`~~, <br/>`0.10.2`, ~~`0.10.1.1`~~, <br/>~~`0.10.0`~~ |                                                                        | `0.0.21`, ~~`0.0.20`~~ |
-| 12.4                         | ~~`0.11.0`~~, ~~`0.10.2`~~, <br/>~~`0.10.1.1`~~, ~~`0.10.0`~~                                     |                                                                        | ~~`0.0.20`~~           |
+| CUDA Version <br/> (Variant) | vLLM                                        | SGLang                                                    | VoxBox   |
+|------------------------------|---------------------------------------------|-----------------------------------------------------------|----------|
+| 12.9                         | `0.13.0`, `0.12.0`, <br/>`0.11.2`           | `0.5.7`, `0.5.6.post2`                                    |          |
+| 12.8                         | `0.13.0`, `0.12.0`, <br/>`0.11.2`, `0.10.2` | `0.5.7`, `0.5.6.post2`, <br/>`0.5.5.post3`, `0.5.4.post3` | `0.0.21` |
+| 12.6                         | `0.13.0`, `0.12.0`, <br/>`0.11.2`,`0.10.2`  |                                                           | `0.0.21` |
 
 ### Hygon DTK
 
@@ -101,6 +89,12 @@ The following table lists the supported accelerated backends and their correspon
 |-----------------------------|----------------------------|
 | 25.04                       | `0.11.0`, `0.9.2`, `0.8.5` |
 
+### THead HGGC
+
+| HGGC Version <br/> (Variant) | vLLM     | SGLang  |
+|------------------------------|----------|---------|
+| 12.3                         | `0.11.1` | `0.5.5` |
+
 ### MetaX MACA
 
 | MACA Version <br/> (Variant) | vLLM     |
@@ -117,29 +111,23 @@ The following table lists the supported accelerated backends and their correspon
 
 ### AMD ROCm
 
-> [!CAUTION]
-> Since v0.1.23:
-> - Deprecated all services for ROCm 6.3.
-> - Deprecated vLLM `0.11.0`.
-
 > [!NOTE]
 > - ROCm 7.0 supports LLVM targets:
     `gfx908 gfx90a gfx942 gfx950 gfx1030 gfx1100 gfx1101 gfx1200 gfx1201 gfx1150 gfx1151`.
-> - ROCm 6.4/6.3 supports LLVM targets:
+> - ROCm 6.4 supports LLVM targets:
     `gfx908 gfx90a gfx942 gfx1030 gfx1100`.
 
 > [!WARNING]
 > - ROCm 7.0 vLLM `0.11.2/0.11.0` are reusing the official ROCm 6.4 PyTorch 2.9 wheel package rather than a ROCm
-    7.0 specific PyTorch build. Although supports ROCm 7.0 in vLLM `0.11.2/0.11.0`, `gfx1150/gfx1151` are not supported yet.
+    7.0 specific PyTorch build. Although supports ROCm 7.0 in vLLM `0.11.2`, `gfx1150/gfx1151` are not supported yet.
 > - ROCm 6.4 vLLM `0.13.0` supports `gfx903 gfx90a gfx942` only.
 > - ROCm 6.4 SGLang supports `gfx942` only.
 > - ROCm 7.0 SGLang supports `gfx950` only.
 
-| ROCm Version <br/> (Variant) | vLLM                                            | SGLang                                |
-|------------------------------|-------------------------------------------------|---------------------------------------|
-| 7.0                          | `0.13.0`, `0.12.0`, <br/>`0.11.2`, ~~`0.11.0`~~ | `0.5.7`, `0.5.6.post2`                |
-| 6.4                          | `0.13.0`, `0.12.0`, <br/>`0.11.2`, `0.10.2`     | `0.5.7`, `0.5.6.post2`, `0.5.5.post3` |
-| 6.3                          | ~~`0.10.1.1`~~, ~~`0.10.0`~~                    |                                       |
+| ROCm Version <br/> (Variant) | vLLM                                        | SGLang                                     |
+|------------------------------|---------------------------------------------|--------------------------------------------|
+| 7.0                          | `0.13.0`, `0.12.0`, <br/>`0.11.2`           | `0.5.7`, `0.5.6.post2`                     |
+| 6.4                          | `0.13.0`, `0.12.0`, <br/>`0.11.2`, `0.10.2` | `0.5.7`, `0.5.6.post2`, <br/>`0.5.5.post3` |
 
 ## Directory Structure
 
@@ -251,7 +239,7 @@ To add support for a new accelerated backend:
 
 1. Create a new directory under `pack/` named with the new backend.
 2. Add a `Dockerfile` in the new directory following the [Dockerfile Convention](#dockerfile-convention).
-3. Update [pack.yml](.github/workflows/pack.yml) to include the new backend in the build matrix.
+3. Update [pack.yml](.github/workflows/pack.yml), [discard.yml](.github/workflows/discard.yml) and [prune.yml](.github/workflows/prune.yml) to include the new backend in the build matrix.
 4. Update [matrix.yml](pack/matrix.yaml) to include the new backend and its variants.
 5. Update `_RE_DOCKER_IMAGE` in [runner.py](gpustack_runner/runner.py) to recognize the new backend.
 6. [Optional] Update [tests](tests/gpustack_runner) if necessary.

{gpustack_runner-0.1.23.post5 → gpustack_runner-0.1.24.post1}/README.md

@@ -26,24 +26,20 @@ The following table lists the supported accelerated backends and their correspon
 
 ### Ascend CANN
 
-> [!CAUTION]
-> Since v0.1.23:
-> - Deprecated MindIE `2.1.rc1`.
-
 > [!WARNING]
 > - The Atlas 300I series is currently experimental in vLLM, only supporting eager mode and float16 data type. And there
     are some known issues for running vLLM, you can refer to
     vllm-ascend [#3316](https://github.com/vllm-project/vllm-ascend/issues/3316)
     and [#2795](https://github.com/vllm-project/vllm-ascend/issues/2795).
 
-| CANN Version <br/> (Variant) | MindIE                   | vLLM                                                       | SGLang                 |
-|------------------------------|--------------------------|------------------------------------------------------------|------------------------|
-| 8.3 (A3/910C)                | `2.2.rc1`                | `0.13.0`, `0.12.0`, `0.11.0`                               | `0.5.7`, `0.5.6.post2` |
-| 8.3 (910B)                   | `2.2.rc1`                | `0.13.0`, `0.12.0`, `0.11.0`                               | `0.5.7`, `0.5.6.post2` |
-| 8.3 (310P)                   | `2.2.rc1`                |                                                            |                        |
-| 8.2 (A3/910C)                | `2.1.rc2`                | `0.10.2`, `0.10.1.1`                                       | `0.5.2`, `0.5.1.post3` |
-| 8.2 (910B)                   | `2.1.rc2`, ~~`2.1.rc1`~~ | `0.10.2`, `0.10.1.1`, <br/>`0.10.0`, `0.9.2`, <br/>`0.9.1` | `0.5.2`, `0.5.1.post3` |
-| 8.2 (310P)                   | `2.1.rc2`, ~~`2.1.rc1`~~ | `0.10.0`, `0.9.2`                                          |                        |
+| CANN Version <br/> (Variant) | MindIE    | vLLM                                                       | SGLang                 |
+|------------------------------|-----------|------------------------------------------------------------|------------------------|
+| 8.3 (A3/910C)                | `2.2.rc1` | `0.13.0`, `0.12.0`, `0.11.0`                               | `0.5.7`, `0.5.6.post2` |
+| 8.3 (910B)                   | `2.2.rc1` | `0.13.0`, `0.12.0`, `0.11.0`                               | `0.5.7`, `0.5.6.post2` |
+| 8.3 (310P)                   | `2.2.rc1` |                                                            |                        |
+| 8.2 (A3/910C)                | `2.1.rc2` | `0.10.2`, `0.10.1.1`                                       | `0.5.2`, `0.5.1.post3` |
+| 8.2 (910B)                   | `2.1.rc2` | `0.10.2`, `0.10.1.1`, <br/>`0.10.0`, `0.9.2`, <br/>`0.9.1` | `0.5.2`, `0.5.1.post3` |
+| 8.2 (310P)                   | `2.1.rc2` | `0.10.0`, `0.9.2`                                          |                        |
 
 ### Iluvatar CoreX
 
@@ -53,13 +49,6 @@ The following table lists the supported accelerated backends and their correspon
 
 ### NVIDIA CUDA
 
-> [!CAUTION]
-> Since v0.1.23:
-> - Deprecated all services for CUDA 12.4.
-> - Deprecated vLLM `0.11.0`, `0.10.1.1`, `0.10.0`.
-> - Deprecated SGLang `0.5.5`.
-> - Deprecated VoxBox `0.0.20`.
-
 > [!NOTE]
 > - CUDA 12.9 supports Compute Capabilities:
     `7.5 8.0+PTX 8.9 9.0 10.0 10.3 12.0 12.1+PTX`.
@@ -68,12 +57,11 @@ The following table lists the supported accelerated backends and their correspon
 > - CUDA 12.6/12.4 supports Compute Capabilities:
     `7.5 8.0+PTX 8.9 9.0+PTX`.
 
-| CUDA Version <br/> (Variant) | vLLM                                                                                              | SGLang                                                                 | VoxBox                 |
-|------------------------------|---------------------------------------------------------------------------------------------------|------------------------------------------------------------------------|------------------------|
-| 12.9                         | `0.13.0`, `0.12.0`, <br/>`0.11.2`                                                                 | `0.5.7`, `0.5.6.post2`                                                 |                        |
-| 12.8                         | `0.13.0`, `0.12.0`, <br/>`0.11.2`, ~~`0.11.0`~~, <br/>`0.10.2`, ~~`0.10.1.1`~~, <br/>~~`0.10.0`~~ | `0.5.7`, `0.5.6.post2`, `0.5.5.post3`, <br/>~~`0.5.5`~~, `0.5.4.post3` | `0.0.21`, ~~`0.0.20`~~ |
-| 12.6                         | `0.13.0`, `0.12.0`, <br/>`0.11.2`, ~~`0.11.0`~~, <br/>`0.10.2`, ~~`0.10.1.1`~~, <br/>~~`0.10.0`~~ |                                                                        | `0.0.21`, ~~`0.0.20`~~ |
-| 12.4                         | ~~`0.11.0`~~, ~~`0.10.2`~~, <br/>~~`0.10.1.1`~~, ~~`0.10.0`~~                                     |                                                                        | ~~`0.0.20`~~           |
+| CUDA Version <br/> (Variant) | vLLM                                        | SGLang                                                    | VoxBox   |
+|------------------------------|---------------------------------------------|-----------------------------------------------------------|----------|
+| 12.9                         | `0.13.0`, `0.12.0`, <br/>`0.11.2`           | `0.5.7`, `0.5.6.post2`                                    |          |
+| 12.8                         | `0.13.0`, `0.12.0`, <br/>`0.11.2`, `0.10.2` | `0.5.7`, `0.5.6.post2`, <br/>`0.5.5.post3`, `0.5.4.post3` | `0.0.21` |
+| 12.6                         | `0.13.0`, `0.12.0`, <br/>`0.11.2`,`0.10.2`  |                                                           | `0.0.21` |
 
 ### Hygon DTK
 
@@ -81,6 +69,12 @@ The following table lists the supported accelerated backends and their correspon
 |-----------------------------|----------------------------|
 | 25.04                       | `0.11.0`, `0.9.2`, `0.8.5` |
 
+### THead HGGC
+
+| HGGC Version <br/> (Variant) | vLLM     | SGLang  |
+|------------------------------|----------|---------|
+| 12.3                         | `0.11.1` | `0.5.5` |
+
 ### MetaX MACA
 
 | MACA Version <br/> (Variant) | vLLM     |
@@ -97,29 +91,23 @@ The following table lists the supported accelerated backends and their correspon
 
 ### AMD ROCm
 
-> [!CAUTION]
-> Since v0.1.23:
-> - Deprecated all services for ROCm 6.3.
-> - Deprecated vLLM `0.11.0`.
-
 > [!NOTE]
 > - ROCm 7.0 supports LLVM targets:
     `gfx908 gfx90a gfx942 gfx950 gfx1030 gfx1100 gfx1101 gfx1200 gfx1201 gfx1150 gfx1151`.
-> - ROCm 6.4/6.3 supports LLVM targets:
+> - ROCm 6.4 supports LLVM targets:
     `gfx908 gfx90a gfx942 gfx1030 gfx1100`.
 
 > [!WARNING]
 > - ROCm 7.0 vLLM `0.11.2/0.11.0` are reusing the official ROCm 6.4 PyTorch 2.9 wheel package rather than a ROCm
-    7.0 specific PyTorch build. Although supports ROCm 7.0 in vLLM `0.11.2/0.11.0`, `gfx1150/gfx1151` are not supported yet.
+    7.0 specific PyTorch build. Although supports ROCm 7.0 in vLLM `0.11.2`, `gfx1150/gfx1151` are not supported yet.
 > - ROCm 6.4 vLLM `0.13.0` supports `gfx903 gfx90a gfx942` only.
 > - ROCm 6.4 SGLang supports `gfx942` only.
 > - ROCm 7.0 SGLang supports `gfx950` only.
 
-| ROCm Version <br/> (Variant) | vLLM                                            | SGLang                                |
-|------------------------------|-------------------------------------------------|---------------------------------------|
-| 7.0                          | `0.13.0`, `0.12.0`, <br/>`0.11.2`, ~~`0.11.0`~~ | `0.5.7`, `0.5.6.post2`                |
-| 6.4                          | `0.13.0`, `0.12.0`, <br/>`0.11.2`, `0.10.2`     | `0.5.7`, `0.5.6.post2`, `0.5.5.post3` |
-| 6.3                          | ~~`0.10.1.1`~~, ~~`0.10.0`~~                    |                                       |
+| ROCm Version <br/> (Variant) | vLLM                                        | SGLang                                     |
+|------------------------------|---------------------------------------------|--------------------------------------------|
+| 7.0                          | `0.13.0`, `0.12.0`, <br/>`0.11.2`           | `0.5.7`, `0.5.6.post2`                     |
+| 6.4                          | `0.13.0`, `0.12.0`, <br/>`0.11.2`, `0.10.2` | `0.5.7`, `0.5.6.post2`, <br/>`0.5.5.post3` |
 
 ## Directory Structure
 
@@ -231,7 +219,7 @@ To add support for a new accelerated backend:
 
 1. Create a new directory under `pack/` named with the new backend.
 2. Add a `Dockerfile` in the new directory following the [Dockerfile Convention](#dockerfile-convention).
-3. Update [pack.yml](.github/workflows/pack.yml) to include the new backend in the build matrix.
+3. Update [pack.yml](.github/workflows/pack.yml), [discard.yml](.github/workflows/discard.yml) and [prune.yml](.github/workflows/prune.yml) to include the new backend in the build matrix.
 4. Update [matrix.yml](pack/matrix.yaml) to include the new backend and its variants.
 5. Update `_RE_DOCKER_IMAGE` in [runner.py](gpustack_runner/runner.py) to recognize the new backend.
 6. [Optional] Update [tests](tests/gpustack_runner) if necessary.

{gpustack_runner-0.1.23.post5 → gpustack_runner-0.1.24.post1}/gpustack_runner/__init__.py

@@ -1,5 +1,11 @@
 from __future__ import annotations
 
+from .__utils__ import (
+    merge_image,
+    parse_image,
+    replace_image_with,
+    split_image,
+)
 from ._version import commit_id, version, version_tuple
 from .runner import (
     BackendRunners,
@@ -21,7 +27,11 @@ __all__ = [
     "list_backend_runners",
     "list_runners",
     "list_service_runners",
+    "merge_image",
+    "parse_image",
+    "replace_image_with",
     "set_re_docker_image",
+    "split_image",
     "version",
     "version_tuple",
 ]

{gpustack_runner-0.1.23.post5 → gpustack_runner-0.1.24.post1}/gpustack_runner/__main__.py

@@ -11,6 +11,7 @@ from .cmds import (
     CompareImagesSubCommand,
     CopyImagesSubCommand,
     ListImagesSubCommand,
+    LoadImagesSubCommand,
     SaveImagesSubCommand,
 )
 
@@ -35,6 +36,7 @@ def main():
     SaveImagesSubCommand.register(subcommand_parser)
     CopyImagesSubCommand.register(subcommand_parser)
     CompareImagesSubCommand.register(subcommand_parser)
+    LoadImagesSubCommand.register(subcommand_parser)
 
     # Autocomplete
     argcomplete.autocomplete(parser)

gpustack_runner-0.1.24.post1/gpustack_runner/__utils__.py

@@ -0,0 +1,155 @@
+from __future__ import annotations
+
+_BLANK_TAG = "latest"
+
+
+def split_image(image: str, fill_blank_tag: bool = False) -> tuple:
+    """
+    Split the Docker completed image string into its image_name([registry/][namespace/]repository) and image_tag.
+
+    Args:
+        image:
+            The Docker completed image string to split.
+        fill_blank_tag:
+            If True, fill the blank tag with `latest`.
+
+    Returns:
+        A tuple of (image_name, image_tag).
+
+    """
+    parts = image.rsplit("@", maxsplit=1)
+    if len(parts) == 2:
+        return tuple(parts)
+    parts = image.rsplit(":", maxsplit=1)
+    if len(parts) == 2 and "/" not in parts[1]:
+        if fill_blank_tag:
+            parts[1] = parts[1] or _BLANK_TAG
+        return tuple(parts)
+    return image, _BLANK_TAG if fill_blank_tag else None
+
+
+def merge_image(image_name: str, image_tag: str | None = None) -> str:
+    """
+    Merge the Docker image and image_tag into a single string.
+
+    Args:
+        image_name:
+            The Docker image name, in form of [registry/][namespace/]repository.
+        image_tag:
+            The Docker image tag.
+
+    Returns:
+        The completed Docker image string.
+
+    """
+    if not image_tag:
+        return image_name
+    if image_tag.startswith("sha256:"):
+        return f"{image_name}@{image_tag}"
+    return f"{image_name}:{image_tag}"
+
+
+def parse_image(
+    image: str,
+    fill_blank_tag: bool = False,
+) -> tuple[str | None, str | None, str, str | None] | None:
+    """
+    Parse the Docker image string into its components:
+    registry, namespace, repository, and tag.
+
+    Args:
+        image:
+            The Docker image string to parse.
+        fill_blank_tag:
+            If True, fill the blank tag with `latest`.
+
+    Returns:
+        A tuple of (registry, namespace, repository, tag).
+        Registry, namespace, and tag can be None if not present.
+        If the image string is invalid, return None.
+
+    """
+    image_reg, image_ns, image_repo, image_tag = (
+        None,
+        None,
+        None,
+        None,
+    )
+    image_rest = image.strip()
+
+    # Get tag.
+    image_rest, image_tag = split_image(image_rest, fill_blank_tag=fill_blank_tag)
+    if not image_rest:
+        return None
+
+    # Get repository.
+    parts = image_rest.rsplit("/", maxsplit=1)
+    if len(parts) == 2:
+        image_rest, image_repo = parts
+    else:
+        image_rest, image_repo = None, image_rest
+
+    # Get namespace.
+    if image_rest:
+        parts = image_rest.rsplit("/", maxsplit=1)
+        if len(parts) == 2:
+            image_reg, image_ns = parts
+        else:
+            image_reg, image_ns = None, image_rest
+
+    return image_reg, image_ns, image_repo, image_tag
+
+
+def replace_image_with(
+    image: str,
+    registry: str | None = None,
+    namespace: str | None = None,
+    repository: str | None = None,
+) -> str:
+    """
+    Replace the registry, namespace, and repository of a Docker image string.
+
+    The given image string is parsed into its components (registry, namespace, repository, tag),
+    and the specified components are replaced with the provided values.
+
+    The format of a Docker image string is:
+    [registry/][namespace/]repository[:tag|@digest]
+
+    Args:
+        image:
+            The original Docker image string.
+        registry:
+            The new registry to use. If None, keep the original registry.
+        namespace:
+            The new namespace to use. If None, keep the original namespace.
+        repository:
+            The new repository to use. If None, keep the original repository.
+
+    Returns:
+        The modified Docker image string.
+
+    """
+    if not image or (not registry and not namespace and not repository):
+        return image
+
+    registry = registry.strip() if registry else None
+    namespace = namespace.strip() if namespace else None
+    repository = repository.strip() if repository else None
+
+    image_reg, image_ns, image_repo, image_tag = parse_image(image)
+
+    registry = registry or image_reg
+    namespace = namespace or image_ns
+    repository = repository or image_repo
+
+    image_name = ""
+    if registry:
+        image_name += f"{registry}/"
+    if namespace:
+        image_name += f"{namespace}/"
+    elif registry:
+        image_name += "library/"
+    image_name += repository
+
+    image = merge_image(image_name, image_tag)
+    return image

{gpustack_runner-0.1.23.post5 → gpustack_runner-0.1.24.post1}/gpustack_runner/_version.py

@@ -27,8 +27,8 @@ version_tuple: VERSION_TUPLE
 __commit_id__: COMMIT_ID
 commit_id: COMMIT_ID
 
-__version__ = version = '0.1.23.post5'
-__version_tuple__ = version_tuple = (0, 1, 23, 'post5')
+__version__ = version = '0.1.24.post1'
+__version_tuple__ = version_tuple = (0, 1, 24, 'post1')
 try:
     from ._version_appendix import git_commit
     __commit_id__ = commit_id = git_commit

gpustack_runner-0.1.24.post1/gpustack_runner/_version_appendix.py

@@ -0,0 +1 @@
+git_commit = "ed41ee9"

{gpustack_runner-0.1.23.post5 → gpustack_runner-0.1.24.post1}/gpustack_runner/cmds/__init__.py

@@ -4,6 +4,7 @@ from .images import (
     CompareImagesSubCommand,
     CopyImagesSubCommand,
     ListImagesSubCommand,
+    LoadImagesSubCommand,
     PlatformedImage,
     SaveImagesSubCommand,
     append_images,
@@ -14,6 +15,7 @@ __all__ = [
     "CompareImagesSubCommand",
     "CopyImagesSubCommand",
     "ListImagesSubCommand",
+    "LoadImagesSubCommand",
     "PlatformedImage",
     "SaveImagesSubCommand",
     "append_images",