don't default to openai compatible tag

basetenlabs · Oct 29, 2024 · 1bf79ee · 1bf79ee
1 parent 26193f7
commit 1bf79ee
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/truss/contexts/image_builder/serving_image_builder.py b/truss/contexts/image_builder/serving_image_builder.py
@@ -26,7 +26,6 @@
     MAX_SUPPORTED_PYTHON_VERSION_IN_CUSTOM_BASE_IMAGE,
     MIN_SUPPORTED_PYTHON_VERSION_IN_CUSTOM_BASE_IMAGE,
     MODEL_DOCKERFILE_NAME,
-    OPENAI_COMPATIBLE_TAG,
     REQUIREMENTS_TXT_FILENAME,
     SERVER_CODE_DIR,
     SERVER_DOCKERFILE_TEMPLATE_NAME,
@@ -421,8 +420,6 @@ def copy_into_build_dir(from_path: Path, path_in_build_dir: str):
                 )
 
                 config.requirements.extend(BASE_TRTLLM_REQUIREMENTS)
-
-                config.model_metadata["tags"] = [OPENAI_COMPATIBLE_TAG]
             else:
                 config.requirements.extend(AUDIO_MODEL_TRTLLM_REQUIREMENTS)
                 config.system_packages.extend(AUDIO_MODEL_TRTLLM_SYSTEM_PACKAGES)