From 7221eecd77ebc6d19eda9ef873a212e7216518ba Mon Sep 17 00:00:00 2001
From: Abhrant <abhiranta@nyunai.com>
Date: Mon, 2 Dec 2024 13:08:58 +0000
Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9Bbug:=20only=20downloading=20nyunzer?=
 =?UTF-8?q?o=20docker=20images?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 zero/core/constants.py | 6 ------
 zero/core/extension.py | 3 +--
 2 files changed, 1 insertion(+), 8 deletions(-)

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'