From d126986fd0f69f5344dba39dd46afde2efa53b72 Mon Sep 17 00:00:00 2001 From: Yunfeng Bai Date: Wed, 6 Mar 2024 00:53:07 +0000 Subject: [PATCH 1/2] Fix cacher --- .../model_engine_server/infra/services/image_cache_service.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model-engine/model_engine_server/infra/services/image_cache_service.py b/model-engine/model_engine_server/infra/services/image_cache_service.py index a14c2b458..c566af0bd 100644 --- a/model-engine/model_engine_server/infra/services/image_cache_service.py +++ b/model-engine/model_engine_server/infra/services/image_cache_service.py @@ -80,7 +80,7 @@ def _cache_finetune_llm_images( ) latest_tag = ( self.docker_repository.get_latest_image_tag( - f"{infra_config().docker_repo_prefix}/{hmi_config.batch_inference_vllm_repository}" + hmi_config.batch_inference_vllm_repository ) if not CIRCLECI else "fake_docker_repository_latest_image_tag" From 7ad700cabdbd75862e326dbc8cb4bf28e805007a Mon Sep 17 00:00:00 2001 From: Yunfeng Bai Date: Wed, 6 Mar 2024 01:02:19 +0000 Subject: [PATCH 2/2] format --- .../model_engine_server/infra/services/image_cache_service.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/model-engine/model_engine_server/infra/services/image_cache_service.py b/model-engine/model_engine_server/infra/services/image_cache_service.py index c566af0bd..5d5c9d135 100644 --- a/model-engine/model_engine_server/infra/services/image_cache_service.py +++ b/model-engine/model_engine_server/infra/services/image_cache_service.py @@ -79,9 +79,7 @@ def _cache_finetune_llm_images( f"{infra_config().docker_repo_prefix}/{hmi_config.vllm_repository}", "0.3.2" ) latest_tag = ( - self.docker_repository.get_latest_image_tag( - hmi_config.batch_inference_vllm_repository - ) + self.docker_repository.get_latest_image_tag(hmi_config.batch_inference_vllm_repository) if not CIRCLECI else "fake_docker_repository_latest_image_tag" )