fix: Update Model Tags (#311)

Signed-off-by: Ishaan Sehgal <ishaanforthewin@gmail.com>
Azure · Mar 26, 2024 · 8eddac3 · 8eddac3
1 parent 99b7d57
commit 8eddac3
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"