Merge pull request #4162 from fred-bf/fix/identify-vision-model
fix: fix the method to detect vision model
This commit is contained in:
commit
8b821ac0c9
|
@ -116,7 +116,7 @@ export class ChatGPTApi implements LLMApi {
|
||||||
enumerable: true,
|
enumerable: true,
|
||||||
configurable: true,
|
configurable: true,
|
||||||
writable: true,
|
writable: true,
|
||||||
value: Math.max(modelConfig.max_tokens, 4096),
|
value: modelConfig.max_tokens,
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -292,8 +292,8 @@ export function getMessageImages(message: RequestMessage): string[] {
|
||||||
|
|
||||||
export function isVisionModel(model: string) {
|
export function isVisionModel(model: string) {
|
||||||
return (
|
return (
|
||||||
model.startsWith("gpt-4-vision") ||
|
// model.startsWith("gpt-4-vision") ||
|
||||||
model.startsWith("gemini-pro-vision") ||
|
// model.startsWith("gemini-pro-vision") ||
|
||||||
!DEFAULT_MODELS.find((m) => m.name == model)
|
model.includes("vision")
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue