fix: vertexai model names (#365)

pull/368/head
sigoden 3 months ago committed by GitHub
parent 7279e43141
commit 4d52479545
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -17,9 +17,9 @@ use std::path::PathBuf;
// https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models
const MODELS: [(&str, usize, &str); 5] = [
("gemini-1.0-pro", 24568, "text"),
("gemini.1.0-pro-vision", 14336, "text,vision"),
("gemini-1.0-pro-vision", 14336, "text,vision"),
("gemini-1.0-ultra", 8192, "text"),
("gemini.1.0-ultra-vision", 8192, "text,vision"),
("gemini-1.0-ultra-vision", 8192, "text,vision"),
("gemini-1.5-pro", 1000000, "text"),
];

Loading…
Cancel
Save