diff --git a/zero/core/constants.py b/zero/core/constants.py index 6a687d3..66c52aa 100644 --- a/zero/core/constants.py +++ b/zero/core/constants.py @@ -148,12 +148,6 @@ def get_env_file_path(workspace_path: Optional[Path]): class DockerRepository(StrEnum): - # kompress - NYUN_KOMPRESS = "nyunadmin/nyun_kompress" - - # adapt - NYUN_ADAPT = "nyunadmin/adapt" - # public NYUN_ZERO_VISION = "nyunadmin/nyunzero_kompress_vision" NYUN_ZERO_TEXT_GENERATION = "nyunadmin/nyunzero_text_generation" diff --git a/zero/core/extension.py b/zero/core/extension.py index 338c31f..581a615 100644 --- a/zero/core/extension.py +++ b/zero/core/extension.py @@ -298,8 +298,7 @@ class AdaptExtension(BaseExtension): extension_type = WorkspaceExtension.ADAPT docker_images = { # public - NyunDocker(DockerRepository.NYUN_ZERO_ADAPT, DockerTag.PUBLIC_LATEST), - NyunDocker(DockerRepository.NYUN_ADAPT, DockerTag.ADAPT), + NyunDocker(DockerRepository.NYUN_ZERO_ADAPT, DockerTag.PUBLIC_LATEST) } extension_metadata = { # huggingface - 'text_classification'