File tree Expand file tree Collapse file tree 1 file changed +5
-1
lines changed
Expand file tree Collapse file tree 1 file changed +5
-1
lines changed Original file line number Diff line number Diff 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+
8589const 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 } ,
You can’t perform that action at this time.
0 commit comments