fix llm-engine finetune.create failures (#464)

scaleapi · Mar 8, 2024 · 659d08d · 659d08d
1 parent 0528b52
commit 659d08d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...gine/model_engine_server/infra/services/docker_image_batch_job_llm_fine_tuning_service.py b/...gine/model_engine_server/infra/services/docker_image_batch_job_llm_fine_tuning_service.py
@@ -77,7 +77,7 @@ async def create_fine_tune(
             raise LLMFineTuningMethodNotImplementedException("Fine-tuning method not accessible")
 
         # TODO: Pass user-defined labels
-        labels = dict(team="egp", product="llm-fine-tune")
+        labels = dict(team="egp", product="training.llm_engine_fine_tune")
 
         logger.info(
             f"Using bundle {di_batch_job_bundle.id} for fine-tune job: {di_batch_job_bundle.image_repository=}, {di_batch_job_bundle.image_tag=}"