Fix up model names that don't begin with 'ggml-'

This commit is contained in:
Adam Treat 2023-06-16 14:43:14 -04:00
parent b39a7d4fd9
commit 782e1e77a4
2 changed files with 3 additions and 2 deletions

View File

@ -412,7 +412,8 @@ QList<QString> Chat::modelList() const
for (const QString& f : fileNames) {
QString filePath = exePath + f;
QFileInfo info(filePath);
QString name = info.completeBaseName().remove(0, 5);
QString basename = info.completeBaseName();
QString name = basename.startsWith("ggml-") ? basename.remove(0, 5) : basename;
if (info.exists()) {
if (name == currentModelName)
list.prepend(name);

View File

@ -83,7 +83,7 @@ Dialog {
id: modelName
objectName: "modelName"
property string filename: modelData.filename
text: !modelData.isChatGPT ? filename.slice(5, filename.length - 4) : filename
text: !modelData.isChatGPT ? (filename.startsWith("ggml-") ? filename.slice(5, filename.length - 4) : filename.slice(0, filename.length - 4)) : filename
padding: 20
anchors.top: parent.top
anchors.left: parent.left