machineuser commited on
Commit
803a700
1 Parent(s): da376c3

Sync widgets demo

Browse files
packages/tasks/src/model-libraries-snippets.ts CHANGED
@@ -538,6 +538,12 @@ export HF_HUB_ENABLE_HF_TRANS: string[]FER=1
538
  huggingface-cli download --local-dir ${nameWithoutNamespace(model.id)} ${model.id}`,
539
  ];
540
 
 
 
 
 
 
 
541
  export const nemo = (model: ModelData): string[] => {
542
  let command: string[] | undefined = undefined;
543
  // Resolve the tag to a nemo domain/sub-domain
 
538
  huggingface-cli download --local-dir ${nameWithoutNamespace(model.id)} ${model.id}`,
539
  ];
540
 
541
+ export const mlxim = (model: ModelData): string[] => [
542
+ `from mlxim.model import create_model
543
+
544
+ model = create_model(${model.id})`,
545
+ ];
546
+
547
  export const nemo = (model: ModelData): string[] => {
548
  let command: string[] | undefined = undefined;
549
  // Resolve the tag to a nemo domain/sub-domain
packages/tasks/src/model-libraries.ts CHANGED
@@ -202,6 +202,15 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
202
  snippets: snippets.mlx,
203
  filter: true,
204
  },
 
 
 
 
 
 
 
 
 
205
  nemo: {
206
  prettyLabel: "NeMo",
207
  repoName: "NeMo",
 
202
  snippets: snippets.mlx,
203
  filter: true,
204
  },
205
+ mlxim: {
206
+ prettyLabel: "mlxim",
207
+ repoName: "mlx-image",
208
+ repoUrl: "https://github.com/riccardomusmeci/mlx-image",
209
+ docsUrl: "https://huggingface.co/docs/hub/mlxim",
210
+ snippets: snippets.mlxim,
211
+ filter: false,
212
+ countDownloads: { term: { path: "model.safetensors" } },
213
+ },
214
  nemo: {
215
  prettyLabel: "NeMo",
216
  repoName: "NeMo",