fixes issue with mistral-large model family not being detected
This commit is contained in:
parent
9ce10b4f6a
commit
8d84f289b2
|
@ -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":
|
||||
|
|
Loading…
Reference in New Issue