diff --git a/server/controller/src/main/java/ai/starwhale/mlops/domain/ft/FineTuneService.java b/server/controller/src/main/java/ai/starwhale/mlops/domain/ft/FineTuneService.java index 6fcc654bf3..b35e8931f7 100644 --- a/server/controller/src/main/java/ai/starwhale/mlops/domain/ft/FineTuneService.java +++ b/server/controller/src/main/java/ai/starwhale/mlops/domain/ft/FineTuneService.java @@ -155,9 +155,9 @@ private FineTuneCreateRequest addEnvToRequest(Long id, FineTuneCreateRequest req ); }).collect(Collectors.joining(" ")); - env.add(new Env("SW_VALIDATION_DATASET_URI", evalDataSetUris)); + env.add(new Env("SW_FINETUNE_VALIDATION_DATASET_URI", evalDataSetUris)); } - env.add(new Env("SW_FINE_TUNE_ID", id.toString())); + env.add(new Env("SW_SERVER_TRIGGERED_FINETUNE_ID", id.toString())); s.setEnv(env); s.verifyStepSpecArgs(); }