Skip to content

Commit 6f22376

Browse files
yagiljulien-c
andauthored
lmstudio deeplink for MLX models (#963)
(Not yet tested) LM Studio 0.3.4 and newer supports MLX - [reference](https://lmstudio.ai/blog/lmstudio-v0.3.4) --------- Co-authored-by: Julien Chaumond <[email protected]>
1 parent e201d55 commit 6f22376

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

packages/tasks/src/local-apps.ts

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,6 +82,10 @@ function isLlamaCppGgufModel(model: ModelData) {
8282
return !!model.gguf?.context_length;
8383
}
8484

85+
function isMlxModel(model: ModelData) {
86+
return model.tags.includes("mlx");
87+
}
88+
8589
const snippetLlamacpp = (model: ModelData, filepath?: string): LocalAppSnippet[] => {
8690
const command = (binary: string) =>
8791
[
@@ -218,7 +222,7 @@ export const LOCAL_APPS = {
218222
prettyLabel: "LM Studio",
219223
docsUrl: "https://lmstudio.ai",
220224
mainTask: "text-generation",
221-
displayOnModelPage: isLlamaCppGgufModel,
225+
displayOnModelPage: (model) => isLlamaCppGgufModel(model) || isMlxModel(model),
222226
deeplink: (model, filepath) =>
223227
new URL(`lmstudio://open_from_hf?model=${model.id}${filepath ? `&file=${filepath}` : ""}`),
224228
},

0 commit comments

Comments
 (0)