modify env

star-whale · Nov 13, 2023 · cb83c7f · cb83c7f
1 parent 75d7ebe
commit cb83c7f
Showing 1 changed file with 2 additions and 2 deletions.
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
@@ -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();
             }