Skip to content

Commit 4083ebd

Browse files
authored
Pass labels to job config (#251)
1 parent 04d445e commit 4083ebd

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

model-engine/model_engine_server/infra/services/docker_image_batch_job_llm_fine_tuning_service.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -73,10 +73,14 @@ async def create_fine_tune(
7373
if not di_batch_job_bundle.public and di_batch_job_bundle.owner != owner:
7474
raise LLMFineTuningMethodNotImplementedException("Fine-tuning method not accessible")
7575

76+
# TODO: Pass user-defined labels
77+
labels = dict(team="egp", product="llm-fine-tune")
78+
7679
batch_job_id = await self.docker_image_batch_job_gateway.create_docker_image_batch_job(
7780
created_by=created_by,
7881
owner=owner,
7982
job_config=dict(
83+
**labels,
8084
gateway_url=os.getenv("GATEWAY_URL"),
8185
user_id=owner,
8286
training_file=training_file,
@@ -97,8 +101,7 @@ async def create_fine_tune(
97101
gpu_type=di_batch_job_bundle.gpu_type,
98102
storage=di_batch_job_bundle.storage,
99103
),
100-
# TODO: Pass user-defined labels
101-
labels=dict(team="egp", product="llm-fine-tune"),
104+
labels=labels,
102105
annotations=dict(fine_tuned_model=fine_tuned_model),
103106
mount_location=di_batch_job_bundle.mount_location,
104107
)

0 commit comments

Comments
 (0)