fixes issue with mistral-large model family not being detected

2024-03-08 17:07:25 -06:00
parent 9ce10b4f6a
commit 8d84f289b2
1 changed files with 1 additions and 1 deletions
@@ -138,7 +138,7 @@ export function getMistralAIModelFamily(model: string): MistralAIModelFamily {
    case "mistral-small":
    case "mistral-medium":
    case "mistral-large":
-      return model as MistralAIModelFamily;
+      return prunedModel as MistralAIModelFamily;
    case "open-mistral-7b":
      return "mistral-tiny";
    case "open-mixtral-8x7b":