fix: upgrade mistral

kaito-project · Feb 28, 2024 · 4d0b9cb · 4d0b9cb
1 parent 57abba5
commit 4d0b9cb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 12 deletions.
diff --git a/pkg/utils/testUtils.go b/pkg/utils/testUtils.go
@@ -4,7 +4,6 @@
 package utils
 
 import (
-	"time"
 	"github.com/aws/karpenter-core/pkg/apis/v1alpha5"
 	"github.com/azure/kaito/api/v1alpha1"
 	appsv1 "k8s.io/api/apps/v1"
@@ -141,15 +140,6 @@ var (
 				},
 			},
 		},
-		{
-			ObjectMeta: metav1.ObjectMeta{
-				Name: "node4",
-				Labels: map[string]string{
-					corev1.LabelInstanceTypeStable: "Standard_NC12s_v3",
-				},
-				DeletionTimestamp: &metav1.Time{Time: time.Now()},
-			},
-		},
 	}
 )
 

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.1",
-		"Mistral7BInstruct":  "0.0.1",
+		"Mistral7B":          "0.0.2",
+		"Mistral7BInstruct":  "0.0.2",
 	}
 
 	baseCommandPresetMistral = "accelerate launch"