fix tuning commands

Signed-off-by: jerryzhuang <[email protected]>
kaito-project · Oct 28, 2024 · 3dbfdb6 · 3dbfdb6
1 parent 1fcd0b1
commit 3dbfdb6
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/presets/models/falcon/model.go b/presets/models/falcon/model.go
@@ -46,7 +46,7 @@ var (
 	}
 
 	baseCommandPresetFalconInference = "accelerate launch"
-	baseCommandPresetFalconTuning    = "python3 metrics_server.py & accelerate launch"
+	baseCommandPresetFalconTuning    = "cd /workspace/tfs/ && python3 metrics_server.py & accelerate launch"
 	falconRunParams                  = map[string]string{
 		"torch_dtype":   "bfloat16",
 		"pipeline":      "text-generation",

diff --git a/presets/models/mistral/model.go b/presets/models/mistral/model.go
@@ -32,7 +32,7 @@ var (
 	}
 
 	baseCommandPresetMistralInference = "accelerate launch"
-	baseCommandPresetMistralTuning    = "python3 metrics_server.py & accelerate launch"
+	baseCommandPresetMistralTuning    = "cd /workspace/tfs/ && python3 metrics_server.py & accelerate launch"
 	mistralRunParams                  = map[string]string{
 		"torch_dtype": "bfloat16",
 		"pipeline":    "text-generation",

diff --git a/presets/models/phi2/model.go b/presets/models/phi2/model.go
@@ -26,7 +26,7 @@ var (
 	}
 
 	baseCommandPresetPhiInference = "accelerate launch"
-	baseCommandPresetPhiTuning    = "python3 metrics_server.py & accelerate launch"
+	baseCommandPresetPhiTuning    = "cd /workspace/tfs/ && python3 metrics_server.py & accelerate launch"
 	phiRunParams                  = map[string]string{
 		"torch_dtype": "float16",
 		"pipeline":    "text-generation",

diff --git a/presets/models/phi3/model.go b/presets/models/phi3/model.go
@@ -44,7 +44,7 @@ var (
 	}
 
 	baseCommandPresetPhiInference = "accelerate launch"
-	baseCommandPresetPhiTuning    = "python3 metrics_server.py & accelerate launch"
+	baseCommandPresetPhiTuning    = "cd /workspace/tfs/ && python3 metrics_server.py & accelerate launch"
 	phiRunParams                  = map[string]string{
 		"torch_dtype":       "auto",
 		"pipeline":          "text-generation",