fix: Update Model Tags (#311)

Signed-off-by: Ishaan Sehgal <[email protected]>
kaito-project · Mar 29, 2024 · 7f4e683 · 7f4e683
1 parent f638608
commit 7f4e683
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/presets/models/falcon/model.go b/presets/models/falcon/model.go
@@ -37,10 +37,10 @@ var (
 	PresetFalcon40BInstructModel = PresetFalcon40BModel + "-instruct"
 
 	PresetFalconTagMap = map[string]string{
-		"Falcon7B":          "0.0.3",
-		"Falcon7BInstruct":  "0.0.3",
-		"Falcon40B":         "0.0.3",
-		"Falcon40BInstruct": "0.0.3",
+		"Falcon7B":          "0.0.4",
+		"Falcon7BInstruct":  "0.0.4",
+		"Falcon40B":         "0.0.5",
+		"Falcon40BInstruct": "0.0.5",
 	}
 
 	baseCommandPresetFalcon = "accelerate launch"

diff --git a/presets/models/mistral/model.go b/presets/models/mistral/model.go
@@ -27,8 +27,8 @@ var (
 	PresetMistral7BInstructModel = PresetMistral7BModel + "-instruct"
 
 	PresetMistralTagMap = map[string]string{
-		"Mistral7B":         "0.0.3",
-		"Mistral7BInstruct": "0.0.3",
+		"Mistral7B":         "0.0.4",
+		"Mistral7BInstruct": "0.0.4",
 	}
 
 	baseCommandPresetMistral = "accelerate launch"

diff --git a/presets/models/phi/model.go b/presets/models/phi/model.go
@@ -22,7 +22,7 @@ var (
 	PresetPhi2Model = "phi-2"
 
 	PresetPhiTagMap = map[string]string{
-		"Phi2": "0.0.2",
+		"Phi2": "0.0.3",
 	}
 
 	baseCommandPresetPhi = "accelerate launch"